diff --git a/modules/models/OpenAI.py b/modules/models/OpenAI.py index 2a2b1fbd..5a6055bd 100644 --- a/modules/models/OpenAI.py +++ b/modules/models/OpenAI.py @@ -25,8 +25,10 @@ def __init__( top_p=1.0, user_name="" ) -> None: + if model_name in MODEL_METADATA: + model_name = MODEL_METADATA[model_name]["model_name"] super().__init__( - model_name=MODEL_METADATA[model_name]["model_name"], + model_name=model_name, temperature=temperature, top_p=top_p, system_prompt=system_prompt,