Method__init__(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, checkpoint=True, disable_self_attn=F
ldm/modules/attention.py:249
Method__init__(self, in_channels, n_heads, d_head,
depth=1, dropout=0., context_dim=None,
ldm/modules/attention.py:299
Method__init__(self, channels, use_conv, dims=2, out_channels=None,padding=1)
ldm/modules/diffusionmodules/openaimodel.py:144
Method__init__(
self,
channels,
emb_channels,
dropout,
out_channels=None,
us
ldm/modules/diffusionmodules/openaimodel.py:180
Method__init__(
self,
image_size,
in_channels,
model_channels,
out_channels,
ldm/modules/diffusionmodules/openaimodel.py:444
Method__init__(
self,
image_size,
in_channels,
model_channels,
out_channels,
ldm/modules/diffusionmodules/openaimodel.py:790
Method__init__(self, *, in_channels, out_channels=None, conv_shortcut=False,
dropout, temb_channels=512)
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
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
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)
ldm/modules/diffusionmodules/model.py:608
Method__init__(self, factor, in_channels, mid_channels, out_channels, depth=2)
ldm/modules/diffusionmodules/model.py:656
Method__init__(self, in_channels, ch, resolution, out_ch, num_res_blocks,
attn_resolutions, dropout=0.0, re
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),
ldm/modules/diffusionmodules/model.py:712
Method__init__(self, in_size, out_size, in_channels, out_channels, ch_mult=2)
ldm/modules/diffusionmodules/model.py:729
Method__init__(self, n_embed, n_layer, vocab_size, max_seq_len=77, device="cuda")
ldm/modules/encoders/modules.py:108
Method__init__(self, model_config, linear_start, linear_end, timesteps=1000, max_noise_level=250, output_size=64,
ldm/modules/encoders/modules.py:480
Method__init__(self, target_dir, input_dir, uid_set_pkl, normal_predict=False, image_size=256, bg_color='white')
ldm/data/teeth_dreamer.py:22
Method__init__(self, image_dir, uid_set_npy, image_size, crop_size, normal_predict=False, single_cond=True)
ldm/data/teeth_dreamer.py:220
Method__init__(self, target_dir, input_dir, validation_dir, test_dir, batch_size, uid_set_pkl, test_set_npy, num_workers=4,
ldm/data/teeth_dreamer.py:279
Method__init__(self, dim, n_heads, d_head, context_dim=None, checkpoint=True)
ldm/models/diffusion/sync_dreamer_attention.py:51
Method__init__(self, volume_dims=(5,16,32,64), zero_123_infer=False, *args, **kwargs)
ldm/models/diffusion/sync_dreamer_attention.py:88
Method__init__(self, time_dim=256, input_dim=128, dims=(32, 64, 128, 256))
ldm/models/diffusion/sync_dreamer_network.py:77
Method__init__(self, in_dim, t_dim, v_dim, dims=(32, 64, 128, 256))
ldm/models/diffusion/sync_dreamer_network.py:154
Method__init__(self, model: Zero123, ddim_num_steps, ddim_discretize="uniform", ddim_eta=1.0, latent_size=32)
ldm/models/diffusion/zero123.py:476
Method__init__(self, time_dim, view_dim, view_num,
input_image_size=256, frustum_volume_depth=48,
ldm/models/diffusion/teeth_dreamer.py:133
Method__init__(self, unet_config, scheduler_config,
finetune_unet=False, finetune_projection=True,
ldm/models/diffusion/teeth_dreamer.py:246
Method__init__(self, model: TeethDreamer, ddim_num_steps, ddim_discretize="uniform", ddim_eta=1.0, latent_size=32)
ldm/models/diffusion/teeth_dreamer.py:620
Method__init__(self, time_dim, view_dim, view_num,
input_image_size=256, frustum_volume_depth=48,
ldm/models/diffusion/sync_dreamer.py:133
Method__init__(self, unet_config, scheduler_config,
finetune_unet=False, finetune_projection=True, use_zero
ldm/models/diffusion/sync_dreamer.py:246
Method__init__(self, model: SyncMultiviewDiffusion, ddim_num_steps, ddim_discretize="uniform", ddim_eta=1.0, latent_size=32)
ldm/models/diffusion/sync_dreamer.py:921
Method__init__(self, input_size, num_layers, mode='ir', drop_ratio=0.4, affine=True)
ldm/thirdp/psp/model_irse.py:12