Update g4f/models.py

kqlio67 · Dec 28, 2024 · 2d8c2ca · 2d8c2ca
1 parent f929022
commit 2d8c2ca
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/g4f/models.py b/g4f/models.py
@@ -22,7 +22,6 @@
     HuggingChat,
     HuggingFace,
     HuggingFace,
-    HuggingSpace,
     Liaobots,
     Mhystical,
     Airforce,
@@ -38,6 +37,11 @@
     ReplicateHome,
     RubiksAI,
     TeachAnything,
+
+    ## HuggingSpace ##
+    BlackForestLabsFlux1Dev,
+    BlackForestLabsFlux1Schnell,
+    VoodoohopFlux1Schnell,
 )
 
 @dataclass(unsafe_hash=True)
@@ -564,13 +568,13 @@ class ImageModel(Model):
 flux_dev = ImageModel(
     name = 'flux-dev',
     base_provider = 'Flux AI',
-    best_provider = IterListProvider([HuggingSpace, HuggingChat, HuggingFace])
+    best_provider = IterListProvider([BlackForestLabsFlux1Dev, HuggingChat, HuggingFace])
 )
 
 flux_schnell = ImageModel(
     name = 'flux-schnell',
     base_provider = 'Flux AI',
-    best_provider = HuggingSpace
+    best_provider = IterListProvider([BlackForestLabsFlux1Schnell, VoodoohopFlux1Schnell])
 )
 
 flux_realism = ImageModel(