From 76fb9eea1e50c8c28073e656920bf6beb3b29d00 Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Fri, 30 Jun 2023 11:55:31 -0700 Subject: [PATCH] use Path / operator --- ollama/engine.py | 2 +- ollama/model.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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