Spaces:
Running
Running
File size: 1,042 Bytes
e0336bc |
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 34 35 36 37 38 39 |
# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
import torch
from easydict import EasyDict
from .shared_config import wan_shared_cfg
# ------------------------ Wan I2V 14B ------------------------#
i2v_14B = EasyDict(__name__="Config: Wan I2V 14B")
i2v_14B.update(wan_shared_cfg)
i2v_14B.i2v = True
i2v_14B.is_fun_control = False
i2v_14B.t5_checkpoint = "models_t5_umt5-xxl-enc-bf16.pth"
i2v_14B.t5_tokenizer = "google/umt5-xxl"
# clip
i2v_14B.clip_model = "clip_xlm_roberta_vit_h_14"
i2v_14B.clip_dtype = torch.float16
i2v_14B.clip_checkpoint = "models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth"
i2v_14B.clip_tokenizer = "xlm-roberta-large"
# vae
i2v_14B.vae_checkpoint = "Wan2.1_VAE.pth"
i2v_14B.vae_stride = (4, 8, 8)
# transformer
i2v_14B.patch_size = (1, 2, 2)
i2v_14B.dim = 5120
i2v_14B.ffn_dim = 13824
i2v_14B.freq_dim = 256
i2v_14B.in_dim = 36
i2v_14B.num_heads = 40
i2v_14B.num_layers = 40
i2v_14B.window_size = (-1, -1)
i2v_14B.qk_norm = True
i2v_14B.cross_attn_norm = True
i2v_14B.eps = 1e-6
|