|
|
| from easydict import EasyDict
|
|
|
| from .shared_config import wan_shared_cfg
|
|
|
|
|
|
|
| ti2v_5B = EasyDict(__name__='Config: Wan TI2V 5B')
|
| ti2v_5B.update(wan_shared_cfg)
|
|
|
|
|
| ti2v_5B.t5_checkpoint = 'models_t5_umt5-xxl-enc-bf16.pth'
|
| ti2v_5B.t5_tokenizer = 'google/umt5-xxl'
|
|
|
|
|
| ti2v_5B.vae_checkpoint = 'Wan2.2_VAE.pth'
|
| ti2v_5B.vae_stride = (4, 16, 16)
|
|
|
|
|
| ti2v_5B.patch_size = (1, 2, 2)
|
| ti2v_5B.dim = 3072
|
| ti2v_5B.ffn_dim = 14336
|
| ti2v_5B.freq_dim = 256
|
| ti2v_5B.num_heads = 24
|
| ti2v_5B.num_layers = 30
|
| ti2v_5B.window_size = (-1, -1)
|
| ti2v_5B.qk_norm = True
|
| ti2v_5B.cross_attn_norm = True
|
| ti2v_5B.eps = 1e-6
|
|
|
|
|
| ti2v_5B.sample_fps = 24
|
| ti2v_5B.sample_shift = 5.0
|
| ti2v_5B.sample_steps = 50
|
| ti2v_5B.sample_guide_scale = 5.0
|
| ti2v_5B.frame_num = 121
|
|
|
