FunctionResnetBlockCausal3D(
dim, kernel_size: Union[int, Tuple[int, int, int]], pad_mode: str = "constant"
)
models/common_modules.py:284
Method__init__(
self,
drift,
diffusion,
*,
t0,
t1,
num_steps,
show-o2/transport/integrators.py:8
Method__init__(
self,
drift,
*,
t0,
t1,
sampler_type,
num_steps,
show-o2/transport/integrators.py:82
Method__init__(self, *, model_type, path_type, loss_type, train_eps, sample_eps, snr_type, do_shift, seq_len)
show-o2/transport/transport.py:43
Method__init__(self,
dim=128,
z_dim=4,
dim_mult=[1, 2, 4, 4],
show-o2/models/wan21_vae.py:267
Method__init__(self,
dim=128,
z_dim=4,
dim_mult=[1, 2, 4, 4],
show-o2/models/wan21_vae.py:371
Method__init__(self,
dim=128,
z_dim=4,
dim_mult=[1, 2, 4, 4],
show-o2/models/wan21_vae.py:485
Method__init__(self, batch_size_t2i, batch_size_lm, batch_size_mmu, S, t2i_image_begin_end=[(128, 1152)], mmu_end=1027, righ
training/omni_attention.py:24