ZengrongLin commited on
Commit
dda9f8d
·
verified ·
1 Parent(s): 7ede583

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +38 -0
config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": {
3
+ "model_paths": "Models/Wan2.2-TI2V-5B"
4
+ },
5
+ "dataset": {
6
+ "dataset_repeat": 1,
7
+ "dataset_num_workers": 8,
8
+ "data_file_keys": "video"
9
+ },
10
+ "action": {
11
+ "action_type": "state_pose"
12
+ },
13
+ "video": {
14
+ "height": 480,
15
+ "width": 640,
16
+ "spatial_division_factor": 32,
17
+ "max_pixels": 16777216,
18
+ "num_frames": 81,
19
+ "num_history_frames": 9
20
+ },
21
+ "training": {
22
+ "learning_rate": 5e-05,
23
+ "seed": 42,
24
+ "max_train_steps": 12000,
25
+ "trainable_models": "dit",
26
+ "weight_decay": 0.01,
27
+ "task": "sft",
28
+ "mixed_precision": "bf16",
29
+ "max_timestep_boundary": 1.0,
30
+ "min_timestep_boundary": 0.0
31
+ },
32
+ "infer": {
33
+ "cfg_scale": 1.0,
34
+ "num_inference_steps": 50,
35
+ "quality": 5,
36
+ "fps": 30
37
+ }
38
+ }