summaryrefslogtreecommitdiffstats
path: root/g4f/locals/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/locals/models.py')
-rw-r--r--g4f/locals/models.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/g4f/locals/models.py b/g4f/locals/models.py
index f82f5448..8e1c06bf 100644
--- a/g4f/locals/models.py
+++ b/g4f/locals/models.py
@@ -37,8 +37,10 @@ def get_model_dir() -> str:
local_dir = os.path.dirname(os.path.abspath(__file__))
project_dir = os.path.dirname(os.path.dirname(local_dir))
model_dir = os.path.join(project_dir, "models")
- if os.path.exists(model_dir):
- return model_dir
+ if not os.path.exists(model_dir):
+ os.mkdir(model_dir)
+ return model_dir
+
def get_models() -> dict[str, dict]:
model_dir = get_model_dir()
@@ -48,4 +50,4 @@ def get_models() -> dict[str, dict]:
else:
models = load_models()
save_models(file_path, models)
- return models \ No newline at end of file
+ return models