Add models configuration
Browse files- models_config.json +36 -0
models_config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"models": {
|
| 3 |
+
"wan21_1b": {
|
| 4 |
+
"hf_repo": "Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
|
| 5 |
+
"config": "train_lora_wan21_1b_24gb.yaml"
|
| 6 |
+
},
|
| 7 |
+
"wan21_14b": {
|
| 8 |
+
"hf_repo": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
|
| 9 |
+
"config": "train_lora_wan21_14b_24gb.yaml"
|
| 10 |
+
},
|
| 11 |
+
"wan22_14b": {
|
| 12 |
+
"hf_repo": "ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16",
|
| 13 |
+
"config": "train_lora_wan22_14b_24gb.yaml"
|
| 14 |
+
},
|
| 15 |
+
"qwen_image": {
|
| 16 |
+
"hf_repo": "Qwen/Qwen-Image",
|
| 17 |
+
"config": "train_lora_qwen_image_24gb.yaml"
|
| 18 |
+
},
|
| 19 |
+
"qwen_image_edit": {
|
| 20 |
+
"hf_repo": "Qwen/Qwen-Image-Edit",
|
| 21 |
+
"config": "train_lora_qwen_image_edit_32gb.yaml"
|
| 22 |
+
},
|
| 23 |
+
"flux_dev": {
|
| 24 |
+
"hf_repo": "black-forest-labs/FLUX.1-dev",
|
| 25 |
+
"config": "train_lora_flux_24gb.yaml"
|
| 26 |
+
},
|
| 27 |
+
"flux_schnell": {
|
| 28 |
+
"hf_repo": "black-forest-labs/FLUX.1-schnell",
|
| 29 |
+
"config": "train_lora_flux_schnell_24gb.yaml"
|
| 30 |
+
}
|
| 31 |
+
},
|
| 32 |
+
"ara_adapters": {
|
| 33 |
+
"wan22_14b": "ostris/accuracy_recovery_adapters/wan22_14b_t2i_torchao_uint4.safetensors",
|
| 34 |
+
"qwen_image": "ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors"
|
| 35 |
+
}
|
| 36 |
+
}
|