Method__init__(
self,
dim=128,
z_dim=4,
dim_mult=[1, 2, 4, 4],
num_res_blocks=2,
wan/modules/vae2_2.py:617
Method__init__(
self,
dim=160,
dec_dim=256,
z_dim=16,
dim_mult=[1, 2, 4, 4],
wan/modules/vae2_2.py:735
Method__init__(
self,
z_dim=48,
c_dim=160,
vae_pth=None,
dim_mult=[1, 2, 4, 4],
wan/modules/vae2_2.py:889
Method__init__(self,
dim=128,
z_dim=4,
dim_mult=[1, 2, 4, 4],
wan/modules/vae2_1.py:266
Method__init__(self,
dim=128,
z_dim=4,
dim_mult=[1, 2, 4, 4],
wan/modules/vae2_1.py:370
Method__init__(self, chan_in, chan_out, kernel_size=3, stride=1, dilation=1, pad_mode="replicate", **kwargs)
wan/modules/animate/face_blocks.py:114
Method__init__(self, in_dim: int, hidden_dim: int, num_heads=int, dtype=None, device=None)
wan/modules/animate/face_blocks.py:130
Method__init__(self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True)
wan/modules/animate/motion_encoder.py:91
Method__init__(self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1, activation=None)
wan/modules/animate/motion_encoder.py:117
Method__init__(
self,
in_channel,
out_channel,
kernel_size,
downsample=False,
blur_kernel=[1, 3, 3, 1],
wan/modules/animate/motion_encoder.py:147
Method__init__(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1])
wan/modules/animate/motion_encoder.py:187
Method__init__(
self,
dim,
num_heads,
window_size=(-1, -1),
qk_norm=True,
ep
wan/modules/animate/model_animate.py:89
Method__init__(self,
patch_size=(1, 2, 2),
text_len=512,
in_dim=36,
wan/modules/animate/model_animate.py:250
Method__init__(
self,
patch_size=(1, 2, 2),
in_dim=16,
dim=2048,
ffn_dim=8192,
wan/modules/s2v/motioner.py:400
Methodcross_attn_ffn(x, context, context_lens, e, crossattn_cache=None,
cross_attn_first_call=None)
wan/modules/model_fast.py:308