Add support for Flux model AlekseyCalvin/AuraFlux_merge_diffusers

Woolverine94 · Jan 5, 2025 · 923fcf2 · 923fcf2
1 parent bd53a5f
commit 923fcf2
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 1 deletion.
diff --git a/ressources/common.py b/ressources/common.py
@@ -1898,7 +1898,8 @@ def is_flux(model):
 (model == "black-forest-labs/FLUX.1-schnell") or \
 (model == "sayakpaul/FLUX.1-merged") or \
 (model == "ChuckMcSneed/FLUX.1-dev") or \
-(model == "enhanceaiteam/Mystic")\
+(model == "enhanceaiteam/Mystic") or \
+(model == "AlekseyCalvin/AuraFlux_merge_diffusers")\
 ):
         is_flux_value = True
     else:

diff --git a/ressources/controlnet.py b/ressources/controlnet.py
@@ -103,6 +103,7 @@
     "sayakpaul/FLUX.1-merged",
     "ChuckMcSneed/FLUX.1-dev",
     "enhanceaiteam/Mystic",
+    "AlekseyCalvin/AuraFlux_merge_diffusers",
     "-[ 🏠 Local models ]-",
 ]
 

diff --git a/ressources/img2img.py b/ressources/img2img.py
@@ -117,6 +117,7 @@
     "sayakpaul/FLUX.1-merged",
     "ChuckMcSneed/FLUX.1-dev",
     "enhanceaiteam/Mystic",
+    "AlekseyCalvin/AuraFlux_merge_diffusers",
     "-[ 🏠 Local models ]-",
 ]
 

diff --git a/ressources/img2img_ip.py b/ressources/img2img_ip.py
@@ -101,6 +101,7 @@
     "sayakpaul/FLUX.1-merged",
     "ChuckMcSneed/FLUX.1-dev",
     "enhanceaiteam/Mystic",
+    "AlekseyCalvin/AuraFlux_merge_diffusers",
     "-[ 🏠 Local models ]-",
 ]
 

diff --git a/ressources/txt2img_sd.py b/ressources/txt2img_sd.py
@@ -118,6 +118,7 @@
     "sayakpaul/FLUX.1-merged",
     "ChuckMcSneed/FLUX.1-dev",
     "enhanceaiteam/Mystic",
+    "AlekseyCalvin/AuraFlux_merge_diffusers",
     "-[ 🏠 Local models ]-",
 ]