diff --git a/g4f/Provider/Airforce.py b/g4f/Provider/Airforce.py index 283d561e586..5200d6f70e8 100644 --- a/g4f/Provider/Airforce.py +++ b/g4f/Provider/Airforce.py @@ -37,7 +37,7 @@ def fetch_imagine_models(cls): default_model = "gpt-4o-mini" default_image_model = "flux" - additional_models_imagine = ["stable-diffusion-xl-base", "stable-diffusion-xl-lightning", "Flux-1.1-Pro"] + additional_models_imagine = ["stable-diffusion-xl-base", "stable-diffusion-xl-lightning", "flux-1.1-pro"] @classmethod def get_models(cls): @@ -87,7 +87,7 @@ def get_models(cls): ### imagine ### "sdxl": "stable-diffusion-xl-base", "sdxl": "stable-diffusion-xl-lightning", - "flux-pro": "Flux-1.1-Pro", + "flux-pro": "flux-1.1-pro", } @classmethod