Skip to content

Commit f01f679

Browse files
authored
Merge pull request xtekky#2424 from ufoptg/patch-1
FIX: Bug fix for flux-pro aka flux-schnell
2 parents f4ff7f8 + 30b9707 commit f01f679

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

g4f/Provider/Airforce.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def fetch_imagine_models(cls):
3737

3838
default_model = "gpt-4o-mini"
3939
default_image_model = "flux"
40-
additional_models_imagine = ["stable-diffusion-xl-base", "stable-diffusion-xl-lightning", "Flux-1.1-Pro"]
40+
additional_models_imagine = ["stable-diffusion-xl-base", "stable-diffusion-xl-lightning", "flux-1.1-pro"]
4141

4242
@classmethod
4343
def get_models(cls):
@@ -87,7 +87,7 @@ def get_models(cls):
8787
### imagine ###
8888
"sdxl": "stable-diffusion-xl-base",
8989
"sdxl": "stable-diffusion-xl-lightning",
90-
"flux-pro": "Flux-1.1-Pro",
90+
"flux-pro": "flux-1.1-pro",
9191
}
9292

9393
@classmethod

0 commit comments

Comments
 (0)