diff --git a/ollama/engine.py b/ollama/engine.py index 91506c4a..9ba01d34 100644 --- a/ollama/engine.py +++ b/ollama/engine.py @@ -30,7 +30,7 @@ def load(model_name, models={}): if not models.get(model_name, None): model_path = path.expanduser(model_name) if not path.exists(model_path): - model_path = path.join(MODELS_CACHE_PATH, model_name + ".bin") + model_path = MODELS_CACHE_PATH / model_name + ".bin" runners = { model_type: cls diff --git a/ollama/model.py b/ollama/model.py index 0c02a25d..c2e2e5dc 100644 --- a/ollama/model.py +++ b/ollama/model.py @@ -7,7 +7,7 @@ from tqdm import tqdm MODELS_MANIFEST = 'https://ollama.ai/api/models' -MODELS_CACHE_PATH = path.join(Path.home(), '.ollama', 'models') +MODELS_CACHE_PATH = Path.home() / '.ollama' / 'models' def models(*args, **kwargs): @@ -78,7 +78,7 @@ def find_bin_file(json_response, location, branch): def download_file(download_url, file_name, file_size): - local_filename = path.join(MODELS_CACHE_PATH, file_name) + '.bin' + local_filename = MODELS_CACHE_PATH / file_name + '.bin' first_byte = path.getsize(local_filename) if path.exists(local_filename) else 0