Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LuChengTHU/dpm-solver
/ functions
Functions
1,626 in github.com/LuChengTHU/dpm-solver
⨍
Functions
1,626
◇
Types & classes
286
Function
get_config
()
examples/score_sde_jax/configs/ve/cifar10_ncsnpp.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/bedroom_ncsnpp_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/celeba_ncsnpp.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/celebahq_256_ncsnpp_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ffhq_ncsnpp_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/cifar10_ncsnpp_continuous.py:21
Function
get_config
()
examples/score_sde_jax/configs/ve/ffhq_256_ncsnpp_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/celebahq_ncsnpp_continuous.py:23
Function
get_config
()
examples/score_sde_jax/configs/ve/cifar10_ddpm.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/church_ncsnpp_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsnv2/bedroom.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsnv2/celeba.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsnv2/cifar10.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/celeba_124.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/cifar10_5.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/celeba_1245.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/cifar10_1245.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/celeba.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/cifar10_124.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/cifar10.py:22
Function
get_config
()
examples/score_sde_jax/configs/ve/ncsn/celeba_5.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/cifar10_ddpmpp_deep_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/cifar10_ncsnpp_deep_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/cifar10_ncsnpp.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/cifar10_ddpmpp_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/cifar10_ddpmpp.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/cifar10_ncsnpp_continuous.py:21
Function
get_config
()
examples/score_sde_jax/configs/vp/ddpm/cifar10_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/ddpm/church.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/ddpm/celebahq.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/ddpm/bedroom.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/ddpm/cifar10_unconditional.py:22
Function
get_config
()
examples/score_sde_jax/configs/vp/ddpm/cifar10.py:22
Function
get_config
()
examples/score_sde_jax/configs/subvp/cifar10_ddpmpp_deep_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/subvp/cifar10_ncsnpp_deep_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/subvp/cifar10_ddpm_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/subvp/cifar10_ddpmpp_continuous.py:22
Function
get_config
()
examples/score_sde_jax/configs/subvp/cifar10_ncsnpp_continuous.py:21
Function
get_custom_cond
(mode)
examples/stable-diffusion/notebook_helpers.py:59
Function
get_data_inverse_scaler
Inverse data normalizer.
examples/score_sde_pytorch/datasets.py:32
Function
get_data_inverse_scaler
Inverse data normalizer.
examples/score_sde_jax/datasets.py:32
Function
get_data_scaler
Data normalizer. Assume data are always in [0, 1].
examples/score_sde_pytorch/datasets.py:23
Function
get_data_scaler
Data normalizer. Assume data are always in [0, 1].
examples/score_sde_jax/datasets.py:23
Function
get_dataset
Create data loaders for training and evaluation. Args: config: A ml_collection.ConfigDict parsed from config files. uniform_dequantization:
examples/score_sde_pytorch/datasets.py:71
Function
get_dataset
Create data loaders for training and evaluation. Args: config: A ml_collection.ConfigDict parsed from config files. additional_dim: An inte
examples/score_sde_jax/datasets.py:71
Function
get_ddpm_params
Get betas and alphas --- parameters used in the original DDPM paper.
examples/score_sde_pytorch/models/utils.py:63
Function
get_ddpm_params
Get betas and alphas --- parameters used in the original DDPM paper.
examples/score_sde_jax/models/utils.py:84
Function
get_dir
Get directory that log files are being written to. will be None if there is no output directory (i.e., if you didn't call start)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:281
Method
get_feature_vectors
(self, x, timesteps, low_res=None, **kwargs)
examples/ddpm_and_guided-diffusion/models/improved_ddpm/unet.py:543
Function
get_inception_model
(inceptionv3=False)
examples/score_sde_pytorch/evaluation.py:35
Function
get_inception_model
(inceptionv3=False)
examples/score_sde_jax/evaluation.py:35
Method
get_last_layer
(self)
examples/stable-diffusion/ldm/modules/diffusionmodules/model.py:364
Function
get_likelihood_fn
Create a function to compute the unbiased log-likelihood estimate of a given data point. Args: sde: A `sde_lib.SDE` object that represents the
examples/score_sde_pytorch/likelihood.py:40
Function
get_likelihood_fn
Create a function to compute the unbiased log-likelihood estimate of a given data point. Args: sde: A `sde_lib.SDE` object that represents the
examples/score_sde_jax/likelihood.py:39
Function
get_logit_fn
Create a logit function for the classifier.
examples/score_sde_jax/models/utils.py:293
Function
get_model
(mode)
examples/stable-diffusion/notebook_helpers.py:52
Method
get_model_output
(x, t)
examples/stable-diffusion/ldm/models/diffusion/plms.py:178
Function
get_network
(config)
examples/score_sde_pytorch/models/ncsnv2.py:31
Function
get_network
(config)
examples/score_sde_jax/models/ncsnv2.py:32
Function
get_optimizer
Returns a flax optimizer object based on `config`.
examples/score_sde_pytorch/losses.py:26
Function
get_optimizer
Returns a flax optimizer object based on `config`.
examples/score_sde_jax/losses.py:28
Function
get_pc_colorizer
Create a image colorization function based on Predictor-Corrector (PC) sampling. Args: sde: An `sde_lib.SDE` object that represents the forward
examples/score_sde_pytorch/controllable_generation.py:85
Function
get_pc_colorizer
Create a image colorization function based on Predictor-Corrector (PC) sampling. Args: sde: An `sde_lib.SDE` object that represents the forward
examples/score_sde_jax/controllable_generation.py:97
Function
get_pc_conditional_sampler
Class-conditional sampling with Predictor-Corrector (PC) samplers. Args: sde: An `sde_lib.SDE` object that represents the forward SDE. scor
examples/score_sde_jax/controllable_generation.py:211
Function
get_pc_inpainter
Create an image inpainting function that uses PC samplers. Args: sde: An `sde_lib.SDE` object that represents the forward SDE. predictor: A
examples/score_sde_pytorch/controllable_generation.py:8
Function
get_pc_inpainter
Create an image inpainting function that uses PC samplers. Args: sde: An `sde_lib.SDE` object that represents the forward SDE. model: A `fl
examples/score_sde_jax/controllable_generation.py:10
Function
get_sampling_fn
Create a sampling function. Args: config: A `ml_collections.ConfigDict` object that contains all configuration information. sde: A `sde_lib
examples/score_sde_pytorch/sampling.py:81
Function
get_sampling_fn
Create a sampling function. Args: config: A `ml_collections.ConfigDict` object that contains all configuration information. sde: A `sde_lib
examples/score_sde_jax/sampling.py:85
Function
get_step_fn
Create a one-step training/evaluation function. Args: sde: An `sde_lib.SDE` object that represents the forward SDE. optimize_fn: An optimiz
examples/score_sde_pytorch/losses.py:151
Function
get_step_fn
Create a one-step training/evaluation function. Args: sde: An `sde_lib.SDE` object that represents the forward SDE. model: A `flax.linen.Mo
examples/score_sde_jax/losses.py:179
Function
get_timestep_embedding
(timesteps, embedding_dim, max_positions=10000)
examples/score_sde_pytorch/models/layers.py:515
Function
get_timestep_embedding
(timesteps, embedding_dim, max_positions=10000)
examples/score_sde_jax/models/layers.py:450
Function
get_weight
Get/create weight tensor for a convolution or fully-connected layer.
examples/score_sde_pytorch/models/up_or_down_sampling.py:14
Method
get_x_prev_and_pred_x0
(e_t, index)
examples/stable-diffusion/ldm/models/diffusion/plms.py:199
Function
getkvs
()
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:243
Function
group_by_key_prefix
(prefix, d)
examples/stable-diffusion/ldm/modules/x_transformer.py:106
Function
hinge_d_loss_with_exemplar_weights
(logits_real, logits_fake, weights)
examples/stable-diffusion/ldm/modules/losses/vqperceptual.py:11
Function
imresize
(img, scale, antialiasing=True)
examples/stable-diffusion/ldm/modules/image_degradation/utils_image.py:766
Function
imresize_np
(img, scale, antialiasing=True)
examples/stable-diffusion/ldm/modules/image_degradation/utils_image.py:839
Function
imsave
(img, img_path)
examples/stable-diffusion/ldm/modules/image_degradation/utils_image.py:203
Function
imshow
(x, title=None, cbar=False, figsize=None)
examples/stable-diffusion/ldm/modules/image_degradation/utils_image.py:37
Function
imwrite
(img, img_path)
examples/stable-diffusion/ldm/modules/image_degradation/utils_image.py:209
Function
info
(*args)
examples/ddpm_and_guided-diffusion/models/guided_diffusion/logger.py:258
Function
init
(shape, dtype=dtype, device=device)
examples/score_sde_pytorch/models/layers.py:66
Function
init_
(tensor)
examples/stable-diffusion/ldm/modules/attention.py:29
Method
init_embed
(key, shape, dtype=jnp.float32)
examples/score_sde_jax/models/normalization.py:122
Function
init_model
Initialize a `flax.linen.Module` model.
examples/score_sde_jax/models/utils.py:109
Function
inner
(*args, **kwargs)
examples/stable-diffusion/ldm/modules/x_transformer.py:65
Method
inner_dtype
Get the dtype used by the torso of the model.
examples/ddpm_and_guided-diffusion/models/improved_ddpm/unet.py:456
Function
inpaint_update_fn
(model, data, mask, x, t)
examples/score_sde_pytorch/controllable_generation.py:44
Function
inpaint_update_fn
(rng, state, data, mask, x, t)
examples/score_sde_jax/controllable_generation.py:49
Method
inverse
Inverse the sample `x` from time `t_start` to `t_end` by DPM-Solver. For discrete-time DPMs, we use `t_start=1/N`, where `N` is the t
dpm_solver_pytorch.py:1032
Method
inverse
Inverse the sample `x` from time `t_start` to `t_end` by DPM-Solver. For discrete-time DPMs, we use `t_start=1/N`, where `N` is the t
examples/ddpm_and_guided-diffusion/dpm_solver/sampler.py:1032
Function
l1
(x, y)
examples/stable-diffusion/ldm/modules/losses/vqperceptual.py:35
Function
l2
(x, y)
examples/stable-diffusion/ldm/modules/losses/vqperceptual.py:39
Function
list_dir
List all directories at a given root Args: root (str): Path to directory whose folders need to be listed prefix (bool, optional):
examples/ddpm_and_guided-diffusion/datasets/utils.py:88
Function
list_files
List all files ending with a suffix at a given root Args: root (str): Path to directory whose folders need to be listed suffix (s
examples/ddpm_and_guided-diffusion/datasets/utils.py:110
Function
load_dataset_stats
Load the pre-computed dataset statistics.
examples/score_sde_pytorch/evaluation.py:43
Function
load_dataset_stats
Load the pre-computed dataset statistics.
examples/score_sde_jax/evaluation.py:43
Function
load_multi_files
(data_archive)
examples/stable-diffusion/scripts/train_searcher.py:36
← previous
next →
1,301–1,400 of 1,626, ranked by callers