File size: 680 Bytes
ebc7f2e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
exp_name: ldf
seed: 1234
debug: false
train: false

save_dir: ./outputs
resume_ckpt: null
test_ckpt: "model.safetensors"
test_vae_ckpt: "vae.safetensors"

test_vae:
    target: ldf_models.vae_wan_1d.VAEWanModel
    ema_decay: 0.99
    params:
        input_dim: 263
        z_dim: 4

test_setting:
    render: false
    simple: true
    recover_dim: 263

val_repeat: 1

model:
    target: ldf_models.diffusion_forcing_wan.DiffForcingWanModel
    ema_decay: 0.99
    params:
        checkpoint_path: "ldf_deps/t5_umt5-xxl-enc-bf16/models_t5_umt5-xxl-enc-bf16.pth"
        tokenizer_path: "ldf_deps/t5_umt5-xxl-enc-bf16/google/umt5-xxl"
        input_dim: 4
        noise_steps: 10