Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.,
relative_position=Fals
lvdm/modules/attention.py:44
Method__init__(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, checkpoint=True,
dis
lvdm/modules/attention.py:214
Method__init__(self, in_channels, n_heads, d_head, depth=1, dropout=0., context_dim=None,
use_checkpoint=Tr
lvdm/modules/attention.py:259
Method__init__(self, in_channels, n_heads, d_head, depth=1, dropout=0., context_dim=None,
use_checkpoint=Tr
lvdm/modules/attention.py:320
Method__init__(
self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.0, **kwargs
)
lvdm/modules/attention_svd.py:349
Method__init__(self, channels, use_conv, dims=2, out_channels=None, padding=1)
lvdm/modules/networks/openaimodel3d.py:60
Method__init__(
self,
channels,
emb_channels,
dropout,
out_channels=None,
us
lvdm/modules/networks/openaimodel3d.py:126
Method__init__(self, *, in_channels, out_channels=None, conv_shortcut=False,
dropout, temb_channels=512)
lvdm/modules/networks/ae_modules.py:154
Method__init__(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks,
attn_resolutions, dropout=0.0, resam
lvdm/modules/networks/ae_modules.py:367
Method__init__(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks,
attn_resolutions, dropout=0.0, resam
lvdm/modules/networks/ae_modules.py:479
Method__init__(self, in_channels, out_channels, ch, num_res_blocks, resolution,
ch_mult=(2,2), dropout=0.0)
lvdm/modules/networks/ae_modules.py:630
Method__init__(self, factor, in_channels, mid_channels, out_channels, depth=2)
lvdm/modules/networks/ae_modules.py:678
Method__init__(self, in_channels, ch, resolution, out_ch, num_res_blocks,
attn_resolutions, dropout=0.0, re
lvdm/modules/networks/ae_modules.py:715
Method__init__(self, z_channels, out_ch, resolution, num_res_blocks, attn_resolutions, ch, ch_mult=(1,2,4,8),
lvdm/modules/networks/ae_modules.py:734
Method__init__(self, in_size, out_size, in_channels, out_channels, ch_mult=2)
lvdm/modules/networks/ae_modules.py:751
Method__init__(self, version="openai/clip-vit-large-patch14", device="cuda", max_length=77,
freeze=True, la
lvdm/modules/encoders/condition.py:97
Method__init__(self, arch="ViT-H-14", version="laion2b_s32b_b79k", device="cuda", max_length=77,
freeze=Tru
lvdm/modules/encoders/condition.py:184
Method__init__(self, arch="ViT-H-14", version="laion2b_s32b_b79k", device="cuda", max_length=77,
freeze=Tru
lvdm/modules/encoders/condition.py:242
Method__init__(self, arch="ViT-H-14", version="laion2b_s32b_b79k", device="cuda",
freeze=True, layer="poole
lvdm/modules/encoders/condition.py:300
Method__init__(self, img_cond_stage_config, image_proj_stage_config, freeze_embedder=True, image_proj_model_trainable=True,
lvdm/models/ddpm3d.py:1042
Method__init__(
self,
*,
in_channels,
out_channels=None,
conv_shortcut=False,
lvdm/models/autoencoder_dualref.py:36
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0, **kwargs)
lvdm/models/autoencoder_dualref.py:258
Method__init__(
self,
out_channels,
*args,
dropout=0.0,
video_kernel_size=3,
lvdm/models/autoencoder_dualref.py:847
Method__init__(self, in_channels, out_channels, video_kernel_size=3, *args, **kwargs)
lvdm/models/autoencoder_dualref.py:915
Method__init__(
self,
*args,
video_kernel_size: Union[int, list] = [3,1,1],
alpha: float
lvdm/models/autoencoder_dualref.py:1124