diff options
| -rw-r--r-- | g4f/locals/models.py | 8 |
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 |