Aloukik21 commited on
Commit
42e02b3
·
verified ·
1 Parent(s): 58881b9

Add models configuration

Browse files
Files changed (1) hide show
  1. 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
+ }