MCPcopy Create free account

hub / github.com/LuChengTHU/dpm-solver / functions

Functions1,626 in github.com/LuChengTHU/dpm-solver

⨍ Functions 1,626◇ Types & classes 286
MethodT
End time of the SDE.
examples/score_sde_pytorch/sde_lib.py:21
MethodT
(self)
examples/score_sde_pytorch/sde_lib.py:90
MethodT
(self)
examples/score_sde_pytorch/sde_lib.py:132
MethodT
(self)
examples/score_sde_pytorch/sde_lib.py:182
MethodT
(self)
examples/score_sde_pytorch/sde_lib.py:223
MethodT
End time of the SDE.
examples/score_sde_jax/sde_lib.py:23
MethodT
(self)
examples/score_sde_jax/sde_lib.py:92
MethodT
(self)
examples/score_sde_jax/sde_lib.py:134
MethodT
(self)
examples/score_sde_jax/sde_lib.py:184
MethodT
(self)
examples/score_sde_jax/sde_lib.py:226
Method__call__
(self, input, target)
examples/ddpm_and_guided-diffusion/datasets/vision.py:63
Method__call__
(self, img)
examples/ddpm_and_guided-diffusion/datasets/__init__.py:21
Method__call__
(self, x, n)
examples/stable-diffusion/scripts/knn2img.py:163
Method__call__
(self, n, **kwargs)
examples/stable-diffusion/ldm/lr_scheduler.py:32
Method__call__
(self, n, **kwargs)
examples/stable-diffusion/ldm/lr_scheduler.py:77
Method__call__
(self, x)
examples/score_sde_jax/models/layers.py:124
Method__call__
(self, x, y)
examples/score_sde_jax/models/layers.py:143
Method__call__
(self, x)
examples/score_sde_jax/models/layers.py:162
Method__call__
(self, x, y)
examples/score_sde_jax/models/layers.py:182
Method__call__
(self, xs)
examples/score_sde_jax/models/layers.py:200
Method__call__
(self, xs, y)
examples/score_sde_jax/models/layers.py:222
Method__call__
(self, xs)
examples/score_sde_jax/models/layers.py:247
Method__call__
(self, xs, y)
examples/score_sde_jax/models/layers.py:282
Method__call__
(self, inputs)
examples/score_sde_jax/models/layers.py:319
Method__call__
(self, inputs)
examples/score_sde_jax/models/layers.py:339
Method__call__
(self, x)
examples/score_sde_jax/models/layers.py:363
Method__call__
(self, x, y)
examples/score_sde_jax/models/layers.py:408
Method__call__
(self, x)
examples/score_sde_jax/models/layers.py:472
Method__call__
(self, x)
examples/score_sde_jax/models/layers.py:501
Method__call__
(self, x)
examples/score_sde_jax/models/layers.py:521
Method__call__
(self, x)
examples/score_sde_jax/models/layers.py:533
Method__call__
(self, x, temb=None, train=True)
examples/score_sde_jax/models/layers.py:552
Method__call__
(self, x, labels, train=True)
examples/score_sde_jax/models/ddpm.py:47
Method__call__
(self, x, time_cond, train=True)
examples/score_sde_jax/models/ncsnpp.py:41
Method__call__
(self, x)
examples/score_sde_jax/models/layerspp.py:39
Method__call__
(self, x, y)
examples/score_sde_jax/models/layerspp.py:51
Method__call__
(self, x)
examples/score_sde_jax/models/layerspp.py:67
Method__call__
(self, x)
examples/score_sde_jax/models/layerspp.py:93
Method__call__
(self, x)
examples/score_sde_jax/models/layerspp.py:122
Method__call__
(self, x, temb=None, train=True)
examples/score_sde_jax/models/layerspp.py:156
Method__call__
(self, x, temb=None, train=True)
examples/score_sde_jax/models/layerspp.py:193
Method__call__
(self, x, labels, train=True)
examples/score_sde_jax/models/ncsnv2.py:50
Method__call__
(self, x, labels, train=True)
examples/score_sde_jax/models/ncsnv2.py:121
Method__call__
(self, x, labels, train=True)
examples/score_sde_jax/models/ncsnv2.py:207
Method__call__
(self, x, labels, train=True)
examples/score_sde_jax/models/ncsnv2.py:283
Method__call__
(self, x)
examples/score_sde_jax/models/wideresnet_noise_conditional.py:235
Method__call__
(self, x, temb=None, train=True)
examples/score_sde_jax/models/wideresnet_noise_conditional.py:249
Method__call__
(self, x, temb=None, train=True)
examples/score_sde_jax/models/wideresnet_noise_conditional.py:289
Method__call__
(self, x, sigmas, train=True)
examples/score_sde_jax/models/wideresnet_noise_conditional.py:304
Method__call__
(self, x)
examples/score_sde_jax/models/normalization.py:54
Method__call__
(self, x)
examples/score_sde_jax/models/normalization.py:70
Method__call__
(self, x)
examples/score_sde_jax/models/normalization.py:91
Method__call__
(self, x, y)
examples/score_sde_jax/models/normalization.py:113
Method__call__
(self, x)
examples/score_sde_jax/models/up_or_down_sampling.py:52
Method__getitem__
(self, i)
examples/ddpm_and_guided-diffusion/evaluate/fid_score.py:82
Method__getitem__
(self, index)
examples/ddpm_and_guided-diffusion/datasets/ffhq.py:31
Method__getitem__
(self, index)
examples/ddpm_and_guided-diffusion/datasets/vision.py:28
Method__getitem__
(self, index)
examples/ddpm_and_guided-diffusion/datasets/lsun.py:38
Method__getitem__
Args: index (int): Index Returns: tuple: Tuple (image, target) where target is the index of the target categ
examples/ddpm_and_guided-diffusion/datasets/lsun.py:146
Method__getitem__
(self, index)
examples/ddpm_and_guided-diffusion/datasets/celeba.py:133
Method__getitem__
(self, idx)
examples/stable-diffusion/main.py:142
Method__getitem__
(self, i)
examples/stable-diffusion/ldm/data/imagenet.py:42
Method__getitem__
(self, i)
examples/stable-diffusion/ldm/data/imagenet.py:339
Method__getitem__
(self, i)
examples/stable-diffusion/ldm/data/lsun.py:39
Method__init__
Create a wrapper class for the forward SDE (VP type). *** Update: We support discrete-time diffusion models by implementing a picewis
dpm_solver_jax.py:8
Method__init__
Construct a DPM-Solver. We support both the noise prediction model ("predicting epsilon") and the data prediction model ("predicting x0").
dpm_solver_jax.py:351
Method__init__
Create a wrapper class for the forward SDE (VP type). *** Update: We support discrete-time diffusion models by implementing a picewis
dpm_solver_pytorch.py:7
Method__init__
Construct a DPM-Solver. We support both DPM-Solver (`algorithm_type="dpmsolver"`) and DPM-Solver++ (`algorithm_type="dpmsolver++"`).
dpm_solver_pytorch.py:338
Method__init__
Create a wrapper class for the forward SDE (VP type). *** Update: We support discrete-time diffusion models by implementing a picewis
examples/ddpm_and_guided-diffusion/dpm_solver/sampler.py:7
Method__init__
Construct a DPM-Solver. We support both DPM-Solver (`algorithm_type="dpmsolver"`) and DPM-Solver++ (`algorithm_type="dpmsolver++"`).
examples/ddpm_and_guided-diffusion/dpm_solver/sampler.py:338
Method__init__
(self, args, config, rank=None)
examples/ddpm_and_guided-diffusion/runners/diffusion.py:120
Method__init__
(self, files, transforms=None)
examples/ddpm_and_guided-diffusion/evaluate/fid_score.py:75
Method__init__
(self, in_channels, pool_features)
examples/ddpm_and_guided-diffusion/evaluate/inception.py:213
Method__init__
(self, in_channels, channels_7x7)
examples/ddpm_and_guided-diffusion/evaluate/inception.py:238
Method__init__
(self, in_channels)
examples/ddpm_and_guided-diffusion/evaluate/inception.py:266
Method__init__
(self, in_channels)
examples/ddpm_and_guided-diffusion/evaluate/inception.py:299
Method__init__
(self, path, transform, resolution=8)
examples/ddpm_and_guided-diffusion/datasets/ffhq.py:9
Method__init__
(self, root, transforms=None, transform=None, target_transform=None)
examples/ddpm_and_guided-diffusion/datasets/vision.py:9
Method__init__
(self, transform=None, target_transform=None)
examples/ddpm_and_guided-diffusion/datasets/vision.py:59
Method__init__
(self, x1, x2, y1, y2)
examples/ddpm_and_guided-diffusion/datasets/__init__.py:15
Method__init__
(self, root, transform=None, target_transform=None)
examples/ddpm_and_guided-diffusion/datasets/lsun.py:12
Method__init__
(self, root, split="train", target_type="attr", transform=N
examples/ddpm_and_guided-diffusion/datasets/celeba.py:50
Method__init__
(self, mu=0.999)
examples/ddpm_and_guided-diffusion/models/ema.py:5
Method__init__
(self, in_channels, with_conv)
examples/ddpm_and_guided-diffusion/models/diffusion.py:37
Method__init__
(self, in_channels, with_conv)
examples/ddpm_and_guided-diffusion/models/diffusion.py:56
Method__init__
(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512)
examples/ddpm_and_guided-diffusion/models/diffusion.py:78
Method__init__
(self, in_channels)
examples/ddpm_and_guided-diffusion/models/diffusion.py:138
Method__init__
( self, spacial_dim: int, embed_dim: int, num_heads_channels: int, out
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:27
Method__init__
(self, channels, use_conv, dims=2, out_channels=None)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:123
Method__init__
( self, channels, emb_channels, dropout, out_channels=None, us
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:160
Method__init__
( self, channels, num_heads=1, num_head_channels=-1, use_checkpoint=Fa
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:267
Method__init__
(self, n_heads)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:333
Method__init__
(self, n_heads)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:366
Method__init__
( self, image_size, in_channels, model_channels, out_channels,
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:427
Method__init__
(self, image_size, in_channels, *args, **kwargs)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:673
Method__init__
( self, image_size, in_channels, model_channels, out_channels,
examples/ddpm_and_guided-diffusion/models/guided_diffusion/unet.py:690
Method__init__
( self, *, model, use_fp16=False, fp16_scale_growth=1e-3, init
examples/ddpm_and_guided-diffusion/models/guided_diffusion/fp16_util.py:149
Method__init__
(self, filename_or_file)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:37
Method__init__
(self, filename)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:99
Method__init__
(self, filename)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:114
← previousnext →701–800 of 1,626, ranked by callers