From f13557f779d2fe0df31d4fba080f21bc73e51b2a Mon Sep 17 00:00:00 2001 From: kqlio67 Date: Sat, 14 Sep 2024 12:52:21 +0300 Subject: Fixing errors in g4f/Provider/Nexra.py --- g4f/Provider/Nexra.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'g4f') 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( -- cgit v1.2.3