H-Liu1997 commited on
Commit
9edb29f
·
verified ·
1 Parent(s): 6effebf

Upload configs/stream_mei.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/stream_mei.yaml +41 -0
configs/stream_mei.yaml ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exp_name: stream_mei
2
+ seed: 1234
3
+
4
+ test_ckpt: '/mnt/data/cpfs/cyy/Projects/FloodDiffusionDev/outputs/20260408_175037_ldf_mei_stream/step_step=40000.ckpt'
5
+ test_vae_ckpt: "${dirs.outputs}/20260309_152855_vae_mei/step_step=200000.ckpt"
6
+ test_vae:
7
+ target: models.vae_wan.VAEWanModel
8
+ ema_decay: 0.99
9
+ params:
10
+ input_dim: 138
11
+ z_dim: 16
12
+ model:
13
+ target: models.diffusion_forcing_wan.DiffForcingWanModel
14
+ ema_decay: 0.99
15
+ params:
16
+ schedule_config:
17
+ noise_type: "linear"
18
+ chunk_size: 5
19
+ steps: 10
20
+ extra_len: 4
21
+ sigma_type: "zero"
22
+ sigma_scale: 1.0
23
+ random_epsilon: 0.05
24
+ text_config:
25
+ len: 512
26
+ dim: 4096
27
+ checkpoint_path: "${dirs.deps}/t5_umt5-xxl-enc-bf16/models_t5_umt5-xxl-enc-bf16.pth"
28
+ tokenizer_path: "${dirs.deps}/t5_umt5-xxl-enc-bf16/google/umt5-xxl"
29
+ input_dim: 16
30
+ mean_path: "${dirs.raw_data}/HumanML3D_138/Mean_TOKENS_20260309_152855_vae_mei_200000.npy"
31
+ std_path: "${dirs.raw_data}/HumanML3D_138/Std_TOKENS_20260309_152855_vae_mei_200000.npy"
32
+ input_keys:
33
+ feature: "token"
34
+ feature_length: "token_length"
35
+ text: "text"
36
+ text_end: "token_text_end"
37
+ cfg_config:
38
+ text_scale: 5.0
39
+ null_scale: -4.0
40
+ prediction_type: "vel"
41
+ causal: False