diff --git a/extensions/openai/models.py b/extensions/openai/models.py index a737f0c6..178ede90 100644 --- a/extensions/openai/models.py +++ b/extensions/openai/models.py @@ -55,7 +55,8 @@ def _load_model(data): setattr(shared.args, k, args[k]) shared.model, shared.tokenizer = load_model(model_name) - + shared.model_name = model_name + # Update shared.settings with custom generation defaults if settings: for k in settings: