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

Upload configs/stream.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/stream.yaml +41 -0
configs/stream.yaml ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exp_name: stream_tiny # Experiment names
2
+ seed: 1234 # Seed value
3
+
4
+ test_ckpt: "${dirs.outputs}/20260220_090205_ldf_stream/step_step=80000.ckpt"
5
+ test_vae_ckpt: "${dirs.outputs}/20260113_233726_vae_wan_z4/step_step=2250000.ckpt"
6
+ test_vae:
7
+ target: models.vae_wan.VAEWanModel
8
+ ema_decay: 0.99
9
+ params:
10
+ input_dim: 263
11
+ z_dim: 4
12
+ model:
13
+ target: models.diffusion_forcing_wan.DiffForcingWanModel
14
+ ema_decay: 0.99
15
+ params:
16
+ schedule_config:
17
+ noise_type: "linear" # linear, exponential
18
+ chunk_size: 5
19
+ steps: 10
20
+ extra_len: 4
21
+ sigma_type: "zero" # zero or memoryless
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: 4
30
+ mean_path: null
31
+ std_path: null
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