Merge pull request #2424 from ufoptg/patch-1

FIX: Bug fix for flux-pro aka flux-schnell
xtekky · Nov 25, 2024 · f01f679 · f01f679
2 parents f4ff7f8 + 30b9707
commit f01f679
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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