MCPcopy Create free account

hub / github.com/ChenWu98/cycle-diffusion / functions

Functions1,414 in github.com/ChenWu98/cycle-diffusion

⨍ Functions 1,414◇ Types & classes 263
↓ 1 callersMethodget_base
(self)
model/lib/latentdiff/ldm/data/imagenet.py:379
↓ 1 callersMethodget_base
(self)
model/lib/stable_diffusion/ldm/data/imagenet.py:379
↓ 1 callersFunctionget_beta_schedule
(*, beta_start, beta_end, num_diffusion_timesteps)
model/lib/ddpm_ddim/utils/diffusion_utils.py:5
↓ 1 callersFunctionget_cond
(mode, selected_path)
model/lib/latentdiff/notebook_helpers.py:107
↓ 1 callersFunctionget_cond
(mode, selected_path)
model/lib/stable_diffusion/notebook_helpers.py:107
↓ 1 callersFunctionget_dataset_splits
(args)
main.py:20
↓ 1 callersMethodget_dir
(self)
model/lib/ddpm_ddim/models/improved_ddpm/logger.py:344
↓ 1 callersMethodget_eval_dataloader
Returns the evaluation :class:`~torch.utils.data.DataLoader`. Subclass and override this method if you want to inject some custom be
trainer/trainer.py:275
↓ 1 callersFunctionget_image_paths
(dataroot)
model/lib/latentdiff/ldm/modules/image_degradation/utils_image.py:67
↓ 1 callersFunctionget_image_paths
(dataroot)
model/lib/stable_diffusion/ldm/modules/image_degradation/utils_image.py:67
↓ 1 callersFunctionget_model
(model)
utils/program_utils.py:4
↓ 1 callersFunctionget_multi_task_dataset_splits
(meta_args, name2dataset_splits)
preprocess/to_model.py:118
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
model/lib/latentdiff/ldm/util.py:81
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
model/lib/stable_diffusion/ldm/util.py:88
↓ 1 callersFunctionget_parser
(**parser_kwargs)
model/lib/stable_diffusion/main.py:24
↓ 1 callersFunctionget_preprocessor
(preprocess_program)
utils/program_utils.py:8
↓ 1 callersFunctionget_rank_without_mpi_import
()
model/lib/ddpm_ddim/models/improved_ddpm/logger.py:359
↓ 1 callersMethodget_test_dataloader
Returns the test :class:`~torch.utils.data.DataLoader`. Subclass and override this method if you want to inject some custom behavior
trainer/trainer.py:304
↓ 1 callersFunctionget_timestamp
()
model/lib/latentdiff/ldm/modules/image_degradation/utils_image.py:33
↓ 1 callersFunctionget_timestamp
()
model/lib/stable_diffusion/ldm/modules/image_degradation/utils_image.py:33
↓ 1 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
model/lib/latentdiff/ldm/modules/diffusionmodules/model.py:12
↓ 1 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
model/lib/stable_diffusion/ldm/modules/diffusionmodules/model.py:12
↓ 1 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
model/lib/ddpm_ddim/models/ddpm/diffusion.py:6
↓ 1 callersMethodget_train_dataloader
Returns the training :class:`~torch.utils.data.DataLoader`. Will use no sampler if :obj:`self.train_dataset` does not implement :obj
trainer/trainer.py:247
↓ 1 callersFunctionget_visualizer
(visualizer_program)
utils/program_utils.py:16
↓ 1 callersMethodget_x_noisy
(self, x, t, noise=None)
model/lib/latentdiff/ldm/models/diffusion/classifier.py:109
↓ 1 callersMethodget_x_noisy
(self, x, t, noise=None)
model/lib/stable_diffusion/ldm/models/diffusion/classifier.py:109
↓ 1 callersFunctiongm_blur_kernel
(mean, cov, size=15)
model/lib/latentdiff/ldm/modules/image_degradation/bsrgan_light.py:86
↓ 1 callersFunctiongm_blur_kernel
(mean, cov, size=15)
model/lib/latentdiff/ldm/modules/image_degradation/bsrgan.py:86
↓ 1 callersFunctiongm_blur_kernel
(mean, cov, size=15)
model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan_light.py:86
↓ 1 callersFunctiongm_blur_kernel
(mean, cov, size=15)
model/lib/stable_diffusion/ldm/modules/image_degradation/bsrgan.py:86
↓ 1 callersFunctioni_DDPM
(dataset_name = 'AFHQ')
model/lib/ddpm_ddim/models/improved_ddpm/script_util.py:102
↓ 1 callersFunctionimread_uint
(path, n_channels=3)
model/lib/latentdiff/ldm/modules/image_degradation/utils_image.py:185
↓ 1 callersFunctionimread_uint
(path, n_channels=3)
model/lib/stable_diffusion/ldm/modules/image_degradation/utils_image.py:185
↓ 1 callersFunctionimssave
imgs: list, N images of size WxHxC
model/lib/latentdiff/ldm/modules/image_degradation/utils_image.py:112
↓ 1 callersFunctionimssave
imgs: list, N images of size WxHxC
model/lib/stable_diffusion/ldm/modules/image_degradation/utils_image.py:112
↓ 1 callersMethodinit_
(self)
model/lib/latentdiff/ldm/modules/x_transformer.py:31
↓ 1 callersMethodinit_
(self)
model/lib/latentdiff/ldm/modules/x_transformer.py:595
↓ 1 callersMethodinit_
(self)
model/lib/stable_diffusion/ldm/modules/x_transformer.py:31
↓ 1 callersMethodinit_
(self)
model/lib/stable_diffusion/ldm/modules/x_transformer.py:595
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
model/lib/latentdiff/ldm/models/autoencoder.py:78
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
model/lib/latentdiff/ldm/models/autoencoder.py:313
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list(), only_model=False)
model/lib/latentdiff/ldm/models/diffusion/classifier.py:70
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
model/lib/stable_diffusion/ldm/models/autoencoder.py:78
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
model/lib/stable_diffusion/ldm/models/autoencoder.py:313
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list(), only_model=False)
model/lib/stable_diffusion/ldm/models/diffusion/classifier.py:70
↓ 1 callersMethodinstantiate_cond_stage
(self, config)
model/lib/latentdiff/ldm/models/diffusion/ddpm.py:502
↓ 1 callersMethodinstantiate_cond_stage
(self, config)
model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:503
↓ 1 callersMethodinstantiate_first_stage
(self, config)
model/lib/latentdiff/ldm/models/diffusion/ddpm.py:495
↓ 1 callersMethodinstantiate_first_stage
(self, config)
model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:496
↓ 1 callersMethodinstantiate_pretrained
(self, config)
model/lib/latentdiff/ldm/modules/diffusionmodules/model.py:807
↓ 1 callersMethodinstantiate_pretrained
(self, config)
model/lib/stable_diffusion/ldm/modules/diffusionmodules/model.py:807
↓ 1 callersFunctionis_image_file
(filename)
model/lib/latentdiff/ldm/modules/image_degradation/utils_image.py:29
↓ 1 callersFunctionis_image_file
(filename)
model/lib/stable_diffusion/ldm/modules/image_degradation/utils_image.py:29
↓ 1 callersMethodis_local_process_zero
Whether or not this process is the local (e.g., on one machine if training in a distributed fashion on several machines) main process
trainer/trainer.py:354
↓ 1 callersFunctionisimage
(x)
model/lib/latentdiff/ldm/util.py:40
↓ 1 callersFunctionisimage
(x)
model/lib/stable_diffusion/ldm/util.py:47
↓ 1 callersMethodkl
(self, other=None)
model/lib/latentdiff/ldm/modules/distributions/distributions.py:39
↓ 1 callersMethodkl
(self, other=None)
model/lib/stable_diffusion/ldm/modules/distributions/distributions.py:39
↓ 1 callersMethodload_classifier
(self, ckpt_path, pool)
model/lib/latentdiff/ldm/models/diffusion/classifier.py:95
↓ 1 callersMethodload_classifier
(self, ckpt_path, pool)
model/lib/stable_diffusion/ldm/models/diffusion/classifier.py:95
↓ 1 callersMethodload_diffusion
(self)
model/lib/latentdiff/ldm/models/diffusion/classifier.py:88
↓ 1 callersMethodload_diffusion
(self)
model/lib/stable_diffusion/ldm/models/diffusion/classifier.py:88
↓ 1 callersFunctionload_model_from_config
(config, sd)
model/lib/latentdiff/sample_diffusion.py:217
↓ 1 callersFunctionload_model_from_config
(config, ckpt)
model/lib/latentdiff/notebook_helpers.py:40
↓ 1 callersFunctionload_model_from_config
(config, ckpt)
model/lib/stable_diffusion/notebook_helpers.py:40
↓ 1 callersFunctionload_model_from_config
(config, ckpt, verbose=False)
model/lib/stable_diffusion/txt2img.py:25
↓ 1 callersMethodlog_images
(self, batch, only_inputs=False, plot_ema=False, **kwargs)
model/lib/latentdiff/ldm/models/autoencoder.py:233
↓ 1 callersMethodlog_images
(self, batch, N=8, n_row=2, sample=True, return_keys=None, **kwargs)
model/lib/latentdiff/ldm/models/diffusion/ddpm.py:372
↓ 1 callersMethodlog_images
(self, batch, only_inputs=False, plot_ema=False, **kwargs)
model/lib/stable_diffusion/ldm/models/autoencoder.py:233
↓ 1 callersMethodlog_images
(self, batch, N=8, n_row=2, sample=True, return_keys=None, **kwargs)
model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:373
↓ 1 callersMethodlog_local
(self, save_dir, split, images, global_step, current_epoch, batch_idx)
model/lib/latentdiff/main.py:321
↓ 1 callersMethodlog_local
(self, save_dir, split, images, global_step, current_epoch, batch_idx)
model/lib/stable_diffusion/main.py:321
↓ 1 callersFunctionlogkv
Log a value of some diagnostic Call this once for each diagnostic quantity, each iteration If called many times, last value will be used.
model/lib/ddpm_ddim/models/improved_ddpm/logger.py:168
↓ 1 callersMethodlogkv
(self, key, val)
model/lib/ddpm_ddim/models/improved_ddpm/logger.py:303
↓ 1 callersFunctionmain
()
main.py:57
↓ 1 callersFunctionmain
()
model/lib/stable_diffusion/txt2img.py:45
↓ 1 callersFunctionmake_beta_schedule
(schedule, n_timestep, linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3)
model/lib/latentdiff/ldm/modules/diffusionmodules/util.py:21
↓ 1 callersFunctionmake_beta_schedule
(schedule, n_timestep, linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3)
model/lib/stable_diffusion/ldm/modules/diffusionmodules/util.py:21
↓ 1 callersMethodmake_cond_schedule
(self, )
model/lib/latentdiff/ldm/models/diffusion/ddpm.py:465
↓ 1 callersMethodmake_cond_schedule
(self, )
model/lib/stable_diffusion/ldm/models/diffusion/ddpm.py:466
↓ 1 callersFunctionmake_convolutional_sample
(model, batch_size, vanilla=False, custom_steps=None, eta=1.0,)
model/lib/latentdiff/sample_diffusion.py:76
↓ 1 callersFunctionmake_convolutional_sample
(batch, model, mode="vanilla", custom_steps=None, eta=1.0, swap_mode=False, masked=False,
model/lib/latentdiff/notebook_helpers.py:206
↓ 1 callersFunctionmake_convolutional_sample
(batch, model, mode="vanilla", custom_steps=None, eta=1.0, swap_mode=False, masked=False,
model/lib/stable_diffusion/notebook_helpers.py:208
↓ 1 callersFunctionmake_convolutional_sample
(model, vanilla, custom_steps, eta, x_T)
model/gan_wrapper/latentdiff_wrapper.py:125
↓ 1 callersFunctionmake_convolutional_sample_conditional
(model, vanilla, custom_steps, eta, x_T, scale, class_label)
model/gan_wrapper/latentdiff_wrapper.py:141
↓ 1 callersFunctionmake_convolutional_sample_with_eps
(model, vanilla, custom_steps, eta, x_T, eps_list, refine_steps)
model/gan_wrapper/latentdiff_stochastic_wrapper.py:115
↓ 1 callersFunctionmake_convolutional_sample_with_eps_conditional
(model, custom_steps, eta, x_T, skip_steps, eps_list, scale
model/gan_wrapper/latentdiff_stochastic_text_wrapper.py:61
↓ 1 callersFunctionmake_convolutional_sample_with_eps_conditional
(model, custom_steps, eta, x_T, skip_steps, eps_list, scale
model/gan_wrapper/stable_diffusion_stochastic_text_wrapper.py:59
↓ 1 callersFunctionmake_convolutional_sample_with_eps_conditional
(model, vanilla, custom_steps, eta, x_T, eps_list, scale, c
model/gan_wrapper/latentdiff_stochastic_wrapper.py:133
↓ 1 callersFunctionmake_output_format
(format, ev_dir, log_suffix="")
model/lib/ddpm_ddim/models/improved_ddpm/logger.py:149
↓ 1 callersMethodmake_schedule
(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)
model/lib/latentdiff/ldm/models/diffusion/plms.py:24
↓ 1 callersMethodmake_schedule
(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)
model/lib/stable_diffusion/ldm/models/diffusion/plms.py:24
↓ 1 callersFunctionmaster_params_to_model_params
Copy the master parameter data back into the model parameters.
model/lib/ddpm_ddim/models/improved_ddpm/fp16_util.py:65
↓ 1 callersFunctionmaster_params_to_state_dict
( model, param_groups_and_shapes, master_params, use_fp16 )
model/lib/ddpm_ddim/models/improved_ddpm/fp16_util.py:95
↓ 1 callersFunctionmax_neg_value
(tensor)
model/lib/latentdiff/ldm/modules/x_transformer.py:82
↓ 1 callersFunctionmax_neg_value
(tensor)
model/lib/stable_diffusion/ldm/modules/x_transformer.py:82
↓ 1 callersFunctionmean_flat
Take the mean over all non-batch dimensions.
model/lib/latentdiff/ldm/modules/diffusionmodules/util.py:193
↓ 1 callersFunctionmean_flat
Take the mean over all non-batch dimensions.
model/lib/stable_diffusion/ldm/modules/diffusionmodules/util.py:193
↓ 1 callersFunctionmeasure_perplexity
(predicted_indices, n_embed)
model/lib/latentdiff/ldm/modules/losses/vqperceptual.py:26
← previousnext →401–500 of 1,414, ranked by callers