Method__init__(
self,
channels,
emb_channels,
dropout,
out_channels=None,
us
examples/ddpm_and_guided-diffusion/models/improved_ddpm/unet.py:122
Method__init__(
self,
in_channels,
model_channels,
out_channels,
num_res_blocks,
examples/ddpm_and_guided-diffusion/models/improved_ddpm/unet.py:301
Method__init__(self, features, n_stages, num_classes, normalizer, act=nn.ReLU())
examples/score_sde_pytorch/models/layers.py:158
Method__init__(self, features, n_blocks, n_stages, num_classes, normalizer, act=nn.ReLU())
examples/score_sde_pytorch/models/layers.py:208
Method__init__(self, in_planes, features, act=nn.ReLU(), start=False, end=False, maxpool=True)
examples/score_sde_pytorch/models/layers.py:278
Method__init__(self, in_planes, features, num_classes, normalizer, act=nn.ReLU(), start=False, end=False)
examples/score_sde_pytorch/models/layers.py:314
Method__init__(self, input_dim, output_dim, kernel_size=3, biases=True, adjust_padding=False)
examples/score_sde_pytorch/models/layers.py:352
Method__init__(self, input_dim, output_dim, num_classes, resample=1, act=nn.ELU(),
normalization=ConditionalI
examples/score_sde_pytorch/models/layers.py:398
Method__init__(self, input_dim, output_dim, resample=None, act=nn.ELU(),
normalization=nn.InstanceNorm2d, adj
examples/score_sde_pytorch/models/layers.py:454
Method__init__(self, act, in_ch, out_ch=None, temb_dim=None, conv_shortcut=False, dropout=0.1)
examples/score_sde_pytorch/models/layers.py:621
Method__init__(self, in_ch=None, out_ch=None, with_conv=False, fir=False,
fir_kernel=(1, 3, 3, 1))
examples/score_sde_pytorch/models/layerspp.py:95
Method__init__(self, in_ch=None, out_ch=None, with_conv=False, fir=False,
fir_kernel=(1, 3, 3, 1))
examples/score_sde_pytorch/models/layerspp.py:130
Method__init__(self, act, in_ch, out_ch=None, temb_dim=None, conv_shortcut=False,
dropout=0.1, skip_rescale=F
examples/score_sde_pytorch/models/layerspp.py:169
Method__init__(self, act, in_ch, out_ch=None, temb_dim=None, up=False, down=False,
dropout=0.1, fir=False, fi
examples/score_sde_pytorch/models/layerspp.py:213
Method__init__(self, in_ch, out_ch, kernel, up=False, down=False,
resample_kernel=(1, 3, 3, 1),
examples/score_sde_pytorch/models/up_or_down_sampling.py:26
Method__init__(self, batch_size, train=None, validation=None, test=None, predict=None,
wrap=False, num_work
examples/stable-diffusion/main.py:163
Method__init__(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config)
examples/stable-diffusion/main.py:241
Method__init__(self, warm_up_steps, lr_min, lr_max, lr_start, max_decay_steps, verbosity_interval=0)
examples/stable-diffusion/ldm/lr_scheduler.py:8
Method__init__(self, warm_up_steps, f_min, f_max, f_start, cycle_lengths, verbosity_interval=0)
examples/stable-diffusion/ldm/lr_scheduler.py:41
Method__init__(self, dim, dim_out=None, mult=4, glu=False, dropout=0.)
examples/stable-diffusion/ldm/modules/x_transformer.py:195
Method__init__(
self,
dim,
depth,
heads=8,
causal=False,
examples/stable-diffusion/ldm/modules/x_transformer.py:371
Method__init__(
self,
*,
num_tokens,
max_seq_len,
attn_layers,
examples/stable-diffusion/ldm/modules/x_transformer.py:549
Method__init__(self, dim, dim_out=None, mult=4, glu=False, dropout=0.)
examples/stable-diffusion/ldm/modules/attention.py:48
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.)
examples/stable-diffusion/ldm/modules/attention.py:153
Method__init__(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, checkpoint=True)
examples/stable-diffusion/ldm/modules/attention.py:197
Method__init__(self, in_channels, n_heads, d_head,
depth=1, dropout=0., context_dim=None)
examples/stable-diffusion/ldm/modules/attention.py:226
Method__init__(self, channels, use_conv, dims=2, out_channels=None,padding=1)
examples/stable-diffusion/ldm/modules/diffusionmodules/openaimodel.py:143
Method__init__(
self,
channels,
emb_channels,
dropout,
out_channels=None,
us
examples/stable-diffusion/ldm/modules/diffusionmodules/openaimodel.py:179
Method__init__(
self,
image_size,
in_channels,
model_channels,
out_channels,
examples/stable-diffusion/ldm/modules/diffusionmodules/openaimodel.py:443
Method__init__(
self,
image_size,
in_channels,
model_channels,
out_channels,
examples/stable-diffusion/ldm/modules/diffusionmodules/openaimodel.py:751
Method__init__(self, *, in_channels, out_channels=None, conv_shortcut=False,
dropout, temb_channels=512)
examples/stable-diffusion/ldm/modules/diffusionmodules/model.py:83