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
Methodupdate_fn
(self, x, t)
examples/score_sde_pytorch/sampling.py:198
Methodupdate_fn
(self, x, t)
examples/score_sde_pytorch/sampling.py:252
Methodupdate_fn
(self, x, t)
examples/score_sde_pytorch/sampling.py:279
Methodupdate_fn
(self, x, t)
examples/score_sde_pytorch/sampling.py:316
Methodupdate_fn
(val)
examples/score_sde_jax/dpm_solver.py:939
Methodupdate_fn
One update of the predictor. Args: rng: A JAX random state. x: A JAX array representing the current state t: A JAX array repres
examples/score_sde_jax/sampling.py:159
Methodupdate_fn
One update of the corrector. Args: rng: A JAX random state. x: A JAX array representing the current state t: A JAX array repres
examples/score_sde_jax/sampling.py:185
Methodupdate_fn
(self, rng, x, t)
examples/score_sde_jax/sampling.py:205
Methodupdate_fn
(self, rng, x, t)
examples/score_sde_jax/sampling.py:259
Methodupdate_fn
(self, rng, x, t)
examples/score_sde_jax/sampling.py:286
Methodupdate_fn
(self, rng, x, t)
examples/score_sde_jax/sampling.py:331
Functionupsample_2d
r"""Upsample a batch of 2D images with the given filter. Accepts a batch of 2D images of the shape `[N, C, H, W]` or `[N, H, W, C]` and upsam
examples/score_sde_pytorch/models/up_or_down_sampling.py:195
Functionupsample_2d
r"""Upsample a batch of 2D images with the given filter. Accepts a batch of 2D images of the shape `[N, C, H, W]` or `[N, H, W, C]` and upsam
examples/score_sde_jax/models/up_or_down_sampling.py:333
Methodvalidation_step
(self, batch, batch_idx)
examples/stable-diffusion/ldm/models/autoencoder.py:164
Methodvalidation_step
(self, batch, batch_idx)
examples/stable-diffusion/ldm/models/autoencoder.py:372
Methodvalidation_step
(self, batch, batch_idx)
examples/stable-diffusion/ldm/models/diffusion/classifier.py:210
Methodvalidation_step
(self, batch, batch_idx)
examples/stable-diffusion/ldm/models/diffusion/ddpm.py:358
Functionwarn
(*args)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:262
Functionworker_init_fn
(_)
examples/stable-diffusion/main.py:146
Methodwritekvs
(self, kvs)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:48
Methodwritekvs
(self, kvs)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:102
Methodwritekvs
(self, kvs)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:119
Methodwritekvs
(self, kvs)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:171
Methodwriteseq
(self, seq)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:84
Functionycbcr2rgb
same as matlab ycbcr2rgb Input: uint8, [0, 255] float, [0, 1]
examples/stable-diffusion/ldm/modules/image_degradation/utils_image.py:553
Functionzero_grad
(model_params)
examples/ddpm_and_guided-diffusion/models/improved_ddpm/fp16_util.py:71
← previous1,601–1,626 of 1,626, ranked by callers