Method__init__(self, *, in_channels, out_channels=None, conv_shortcut=False,
dropout, temb_channels=512)
pixel_generator/ldm/modules/diffusionmodules/model.py:83
Method__init__(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks,
attn_resolutions, dropout=0.0, resam
pixel_generator/ldm/modules/diffusionmodules/model.py:369
Method__init__(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks,
attn_resolutions, dropout=0.0, resam
pixel_generator/ldm/modules/diffusionmodules/model.py:463
Method__init__(self, in_channels, out_channels, ch, num_res_blocks, resolution,
ch_mult=(2,2), dropout=0.0)
pixel_generator/ldm/modules/diffusionmodules/model.py:608
Method__init__(self, factor, in_channels, mid_channels, out_channels, depth=2)
pixel_generator/ldm/modules/diffusionmodules/model.py:656
Method__init__(self, in_channels, ch, resolution, out_ch, num_res_blocks,
attn_resolutions, dropout=0.0, re
pixel_generator/ldm/modules/diffusionmodules/model.py:693
Method__init__(self, z_channels, out_ch, resolution, num_res_blocks, attn_resolutions, ch, ch_mult=(1,2,4,8),
pixel_generator/ldm/modules/diffusionmodules/model.py:712
Method__init__(self, in_size, out_size, in_channels, out_channels, ch_mult=2)
pixel_generator/ldm/modules/diffusionmodules/model.py:729
Method__init__(self, n_embed, n_layer, vocab_size, max_seq_len=77, device="cuda")
pixel_generator/ldm/modules/encoders/modules.py:36
Method__init__(self, version='ViT-L/14', device="cuda", max_length=77, n_repeat=1, normalize=True)
pixel_generator/ldm/modules/encoders/modules.py:140
Method__init__(self, disc_start, codebook_weight=1.0, pixelloss_weight=1.0,
disc_num_layers=3, disc_in_chan
pixel_generator/ldm/modules/losses/vqperceptual.py:44
Method__init__(self, disc_start, logvar_init=0.0, kl_weight=1.0, pixelloss_weight=1.0,
disc_num_layers=3, d
pixel_generator/ldm/modules/losses/contperceptual.py:8
Method__init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
pixel_generator/dit/models.py:29
Method__init__(
self,
*,
betas,
model_mean_type,
model_var_type,
loss_type
pixel_generator/dit/diffusion/gaussian_diffusion.py:153
Method__init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
pixel_generator/mage/models_mage.py:32
Method__init__(self, img_size=256, patch_size=16, in_chans=3,
embed_dim=1024, depth=24, num_heads=16,
pixel_generator/mage/models_mage.py:164
Method__init__(self, *, in_channels, out_channels=None, conv_shortcut=False,
dropout, temb_channels=512)
pixel_generator/mage/taming/modules/diffusionmodules/model.py:79
Method__init__(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks,
attn_resolutions, dropout=0.0, resam
pixel_generator/mage/taming/modules/diffusionmodules/model.py:232
Method__init__(self, num_hiddens, embedding_dim, n_embed, straight_through=True,
kl_weight=5e-4, temp_init=
pixel_generator/mage/taming/modules/vqvae/quantize.py:117
Method__init__(self, n_e, e_dim, beta, remap=None, unknown_index="random",
sane_index_shape=False, legacy=T
pixel_generator/mage/taming/modules/vqvae/quantize.py:221
Method__init__(self, n_embed, embedding_dim, beta, decay=0.99, eps=1e-5,
remap=None, unknown_index="random")
pixel_generator/mage/taming/modules/vqvae/quantize.py:362
Method__init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
pretrained_enc/dino/vits.py:67
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0.,
dr
pretrained_enc/dino/vits.py:94
Method__init__(self, img_size=[224], patch_size=16, in_chans=3, num_classes=0, embed_dim=768, depth=12,
num
pretrained_enc/dino/vits.py:134
Method__init__(self, in_dim, use_bn=False, norm_last_layer=True, nlayers=3, hidden_dim=2048, bottleneck_dim=256)
pretrained_enc/dino/vits.py:253
Method__init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
pretrained_enc/ibot/vits.py:63
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0.,
attn_drop=0., dr
pretrained_enc/ibot/vits.py:90
Method__init__(self, img_size=[224], patch_size=16, in_chans=3, num_classes=0, embed_dim=768, depth=12,
num
pretrained_enc/ibot/vits.py:141