From 30b9707657c66e280e66b43dbd61a3688260edf3 Mon Sep 17 00:00:00 2001 From: TrueSaiyan Date: Mon, 25 Nov 2024 21:28:50 +0800 Subject: FIX: Bug fix for flux-pro aka flux-schnell https://api.airforce/imagine2?model=flux-4o https://api.airforce/imagine2?model=flux-schnell and https://api.airforce/imagine2?model=flux-1.1-pro all generate same images but "Flux-1.1-Pro" doesnt work --- g4f/Provider/Airforce.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'g4f/Provider/Airforce.py') diff --git a/g4f/Provider/Airforce.py b/g4f/Provider/Airforce.py index f5bcfefa..c0799a43 100644 --- a/g4f/Provider/Airforce.py +++ b/g4f/Provider/Airforce.py @@ -36,7 +36,7 @@ class Airforce(AsyncGeneratorProvider, ProviderModelMixin): 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): @@ -86,7 +86,7 @@ class Airforce(AsyncGeneratorProvider, ProviderModelMixin): ### imagine ### "sdxl": "stable-diffusion-xl-base", "sdxl": "stable-diffusion-xl-lightning", - "flux-pro": "Flux-1.1-Pro", + "flux-pro": "flux-1.1-pro", } @classmethod -- cgit v1.2.3