FrancisRing commited on
Commit
192a0e4
·
verified ·
1 Parent(s): 1487835

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +19 -0
config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "i2v": true,
3
+ "use_audio": true,
4
+ "random_prefix_frames": true,
5
+ "sp_size": 1,
6
+ "text_encoder_path": "/tmp/pretrained_models/Wan2.1-T2V-14B/models_t5_umt5-xxl-enc-bf16.pth",
7
+ "image_encoder_path": "None",
8
+ "dit_path": "/tmp/pretrained_models/Wan2.1-T2V-14B/diffusion_pytorch_model-00001-of-00006.safetensors,/tmp/pretrained_models/Wan2.1-T2V-14B/diffusion_pytorch_model-00002-of-00006.safetensors,/tmp/pretrained_models/Wan2.1-T2V-14B/diffusion_pytorch_model-00003-of-00006.safetensors,/tmp/pretrained_models/Wan2.1-T2V-14B/diffusion_pytorch_model-00004-of-00006.safetensors,/tmp/pretrained_models/Wan2.1-T2V-14B/diffusion_pytorch_model-00005-of-00006.safetensors,/tmp/pretrained_models/Wan2.1-T2V-14B/diffusion_pytorch_model-00006-of-00006.safetensors",
9
+ "model_config": {
10
+ "in_dim": 33,
11
+ "audio_hidden_size": 32
12
+ },
13
+ "train_architecture": "lora",
14
+ "lora_target_modules": "q,k,v,o,ffn.0,ffn.2",
15
+ "init_lora_weights": "kaiming",
16
+ "lora_rank": 128,
17
+ "lora_alpha": 64.0
18
+ }
19
+