Merge pull request #1389 from bghira/features/pony-v7

Auraflow training support

Author: bghira

README.md

@@ -198,6 +198,14 @@ HiDream has not been tested on 16G cards, but with aggressive quantisation and p
 
 Flux prefers being trained with multiple large GPUs but a single 16G card should be able to do it with quantisation of the transformer and text encoders.
 
+### Auraflow
+
+- A100-80G (Full tune with DeepSpeed)
+- A100-40G (LoRA, LoKr)
+- 3090 24G (LoRA, LoKr)
+- 4060 Ti 16G, 4070 Ti 16G, 3080 16G (int8, LoRA, LoKr)
+- 4070 Super 12G, 3080 10G, 3060 12GB (nf4, LoRA, LoKr)
+
 ### SDXL, 1024px
 
 - A100-80G (EMA, large batches, LoRA @ insane batch sizes)

configure.py

@@ -29,6 +29,7 @@
         "deepfloyd",
         "omnigen",
         "hidream",
+        "auraflow",
     ],
     "lora": [
         "flux",
@@ -40,6 +41,7 @@
         "ltxvideo",
         "wan",
         "deepfloyd",
+        "auraflow",
     ],
     "controlnet": ["sdxl", "sd1x", "sd2x"],
 }
@@ -56,7 +58,10 @@
     "sana": "terminusresearch/sana-1.6b-1024px",
     "ltxvideo": "Lightricks/LTX-Video",
     "wan": "Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
-    "hidream": "HiDream-ai/HiDream-I1-Full"
+    "hidream": "HiDream-ai/HiDream-I1-Full",
+    "auraflow": "terminusresearch/auraflow-v0.3",
+    "deepfloyd": "DeepFloyd/DeepFloyd-IF-I-XL-v1.0",
+    "omnigen": "Shitao/OmniGen-v1-diffusers",    
 }
 
 default_cfg = {
@@ -70,6 +75,10 @@
     "hidream": 2.5,
     "wan": 4.0,
     "sana": 3.8,
+    "omnigen": 3.2,
+    "deepfloyd": 6.0,
+    "sd2x": 7.0,
+    "sd1x": 6.0,    
 }
 
 model_labels = {