diff options
| -rw-r--r-- | g4f/Provider/Nexra.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/g4f/Provider/Nexra.py b/g4f/Provider/Nexra.py index 65c50e73..b2b83837 100644 --- a/g4f/Provider/Nexra.py +++ b/g4f/Provider/Nexra.py @@ -56,17 +56,14 @@ class Nexra(AsyncGeneratorProvider, ProviderModelMixin): "dalle-2": "dalle2", } - @classmethod def get_model(cls, model: str) -> str: if model in cls.text_models or model in cls.image_models: return model elif model in cls.model_aliases: return cls.model_aliases[model] - elif model in cls.image_models: - return cls.default_image_model else: - return cls.default_chat_model + return cls.default_model @classmethod async def create_async_generator( |