Method__init__(self, pdc, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=False)
tools/annotator/sketch/pidinet.py:413
Method__init__(self, arch="ViT-H-14", pretrained="laion2b_s32b_b79k", device="cuda", max_length=77,
freeze
tools/videocomposer/inference_single.py:103
Method__init__(self, arch="ViT-H-14", pretrained="laion2b_s32b_b79k", device="cuda", max_length=77,
freeze
tools/videocomposer/inference_single.py:249
Method__init__(self, dim, context_dim=None, num_heads=None, head_dim=None, batch_size=4)
tools/videocomposer/mha_flash.py:15
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.0)
tools/videocomposer/unet_sd.py:203
Method__init__(self, in_channels, n_heads, d_head,
depth=1, dropout=0., context_dim=None,
tools/videocomposer/unet_sd.py:298
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.)
tools/videocomposer/unet_sd.py:357
Method__init__(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, checkpoint=True,
d
tools/videocomposer/unet_sd.py:412
Method__init__(self, in_dim, embed_dim, out_dim, use_scale_shift_norm=True,
mode='none', dropout=0.0)
tools/videocomposer/unet_sd.py:674
Method__init__(self, in_channels, n_heads, d_head,
depth=1, dropout=0., context_dim=None,
tools/videocomposer/unet_sd.py:894
Method__init__(self, heads=8, dim=2048, dim_head_k=256, dim_head_v=256, dropout_atte = 0.05, mlp_dim=2048, dropout_ffn = 0.0
tools/videocomposer/unet_sd.py:1812
Method__init__(self, arch="ViT-H-14", pretrained="laion2b_s32b_b79k", device="cuda", max_length=77,
freeze
tools/videocomposer/inference_multi.py:103
Method__init__(self, arch="ViT-H-14", pretrained="laion2b_s32b_b79k", device="cuda", max_length=77,
freeze
tools/videocomposer/inference_multi.py:249
Method__init__(self, *, in_channels, out_channels=None, conv_shortcut=False,
dropout, temb_channels=512)
tools/videocomposer/autoencoder.py:77
Method__init__(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks,
attn_resolutions, dropout=0.0, resam
tools/videocomposer/autoencoder.py:378
Method__init__(self, optimizer, base_lr, warmup_steps, total_steps, decay_mode='cosine', min_lr=0.0, last_step=-1)
artist/optim/lr_scheduler.py:8
Method__init__(self, dataset_size, batch_size, num_replicas=None, rank=None, shuffle=False, seed=None)
artist/data/samplers.py:14
Method__init__(self, group_file, batch_size, alpha=0.7, update_interval=5000, seed=8888)
artist/data/samplers.py:42
Method__init__(self, group_file, batch_size, alpha=0.7, update_interval=5000, seed=8888)
artist/data/samplers.py:81