| { |
| "models": { |
| "wan21_1b": { |
| "hf_repo": "Wan-AI/Wan2.1-T2V-1.3B-Diffusers", |
| "config": "train_lora_wan21_1b_24gb.yaml" |
| }, |
| "wan21_14b": { |
| "hf_repo": "Wan-AI/Wan2.1-T2V-14B-Diffusers", |
| "config": "train_lora_wan21_14b_24gb.yaml" |
| }, |
| "wan22_14b": { |
| "hf_repo": "ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16", |
| "config": "train_lora_wan22_14b_24gb.yaml" |
| }, |
| "qwen_image": { |
| "hf_repo": "Qwen/Qwen-Image", |
| "config": "train_lora_qwen_image_24gb.yaml" |
| }, |
| "qwen_image_edit": { |
| "hf_repo": "Qwen/Qwen-Image-Edit", |
| "config": "train_lora_qwen_image_edit_32gb.yaml" |
| }, |
| "flux_dev": { |
| "hf_repo": "black-forest-labs/FLUX.1-dev", |
| "config": "train_lora_flux_24gb.yaml" |
| }, |
| "flux_schnell": { |
| "hf_repo": "black-forest-labs/FLUX.1-schnell", |
| "config": "train_lora_flux_schnell_24gb.yaml" |
| } |
| }, |
| "ara_adapters": { |
| "wan22_14b": "ostris/accuracy_recovery_adapters/wan22_14b_t2i_torchao_uint4.safetensors", |
| "qwen_image": "ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors" |
| } |
| } |
