MCPcopy Create free account

hub / github.com/horseee/DeepCache / functions

Functions1,211 in github.com/horseee/DeepCache

↓ 1 callersMethodq_mean_variance
Get the distribution q(x_t | x_0). :param x_start: the [N x C x ...] tensor of noiseless inputs. :param t: the number of diff
experiments/ldm/ldm/models/diffusion/ddpm.py:204
↓ 1 callersFunctionrandom_crop
(lq, hq, sf=4, lq_patchsize=64)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:431
↓ 1 callersMethodregister_modules
(self, **kwargs)
DeepCache/svd/pipeline_utils.py:556
↓ 1 callersMethodregister_modules
(self, **kwargs)
DeepCache/sd/pipeline_utils.py:507
↓ 1 callersFunctionremove_batch_counter_hook_function
(module)
DeepCache/flops.py:518
↓ 1 callersFunctionrescale_noise_cfg
Rescale `noise_cfg` according to `guidance_rescale`. Based on findings of [Common Diffusion Noise Schedules and Sample Steps are Flawed](http
DeepCache/sdxl/pipeline_stable_diffusion_xl.py:75
↓ 1 callersFunctionrescale_noise_cfg
Rescale `noise_cfg` according to `guidance_rescale`. Based on findings of [Common Diffusion Noise Schedules and Sample Steps are Flawed](http
DeepCache/sdxl/pipeline_stable_diffusion_xl_img2img.py:74
↓ 1 callersFunctionrescale_noise_cfg
Rescale `noise_cfg` according to `guidance_rescale`. Based on findings of [Common Diffusion Noise Schedules and Sample Steps are Flawed](http
DeepCache/sd/pipeline_stable_diffusion.py:98
↓ 1 callersMethodreset_current_t
(self)
experiments/ldm/ldm/models/diffusion/ddpm.py:1433
↓ 1 callersMethodreset_noise_accs
(self)
experiments/ldm/ldm/models/diffusion/classifier.py:202
↓ 1 callersFunctionrun
(model, logdir, batch_size=50, vanilla=False, custom_steps=None, eta=None, n_samples=50000, nplog=None)
experiments/ldm/scripts/sample_diffusion.py:108
↓ 1 callersMethodrun_safety_checker
(self, image, device, dtype)
DeepCache/sd/pipeline_stable_diffusion.py:456
↓ 1 callersMethodsample
(self, cond, batch_size=16, return_intermediates=False, x_T=None, verbose=True, timesteps=None,
experiments/ldm/ldm/models/diffusion/ddpm.py:1217
↓ 1 callersMethodsample_fid
(self, model, total_n_samples=50000, save_images = True)
experiments/ddpm/ddpm/runners/diffusion.py:293
↓ 1 callersMethodsample_fid
(self, model, total_n_samples=50000, save_images = True, timesteps=None)
experiments/ddpm/ddpm/runners/deepcache.py:159
↓ 1 callersMethodsample_from_quad
(self, total_numbers, n_samples, pow)
experiments/ldm/ldm/models/diffusion/ddpm.py:1406
↓ 1 callersFunctionsample_from_quad_center
(total_numbers, n_samples, center, pow=1.2)
DeepCache/sdxl/pipeline_stable_diffusion_xl.py:88
↓ 1 callersFunctionsample_from_quad_center
(total_numbers, n_samples, center, pow=1.2)
DeepCache/sdxl/pipeline_stable_diffusion_xl_img2img.py:87
↓ 1 callersFunctionsample_from_quad_center
(total_numbers, n_samples, center, pow=1.2)
DeepCache/sd/pipeline_stable_diffusion.py:86
↓ 1 callersFunctionsample_from_quad_center
(total_numbers, n_samples, center, pow=1.2)
DeepCache/sd/pipeline_text_to_video_zero.py:46
↓ 1 callersFunctionsample_from_quad_center
(total_numbers, n_samples, center, pow=1.2)
experiments/ddpm/ddpm/functions/deepcache_denoising.py:19
↓ 1 callersMethodsample_image
(self, x, model, last=True, timesteps=None)
experiments/ddpm/ddpm/runners/deepcache.py:202
↓ 1 callersMethodsample_interpolation
(self, model)
experiments/ddpm/ddpm/runners/diffusion.py:352
↓ 1 callersMethodsample_sequence
(self, model)
experiments/ddpm/ddpm/runners/diffusion.py:329
↓ 1 callersFunctionsave_fid_stats
Calculates the FID of two paths
experiments/ddpm/fid.py:305
↓ 1 callersFunctionsave_logs
(logs, path, n_saved=0, key="sample", np_path=None)
experiments/ldm/scripts/sample_diffusion.py:143
↓ 1 callersMethodschedule
(self, n, **kwargs)
experiments/ldm/ldm/lr_scheduler.py:17
↓ 1 callersMethodschedule
(self, n, **kwargs)
experiments/ldm/ldm/lr_scheduler.py:59
↓ 1 callersMethodsearch
(self, x, k)
experiments/ldm/scripts/knn2img.py:135
↓ 1 callersFunctionsearch_ah
(searcher, dims_per_block, aiq_threshold, reorder_k)
experiments/ldm/scripts/train_searcher.py:24
↓ 1 callersFunctionsearch_bruteforce
(searcher)
experiments/ldm/scripts/train_searcher.py:12
↓ 1 callersFunctionsearch_partioned_ah
(searcher, dims_per_block, aiq_threshold, reorder_k, partioning_trainsize, num_leaves,
experiments/ldm/scripts/train_searcher.py:16
↓ 1 callersMethodset_attention_slice
r""" Enable sliced attention computation. When this option is enabled, the attention module splits the input tensor in slices to comp
DeepCache/sdxl/unet_2d_condition.py:670
↓ 1 callersMethodset_attention_slice
(self, slice_size: Optional[int])
DeepCache/sdxl/pipeline_utils.py:1831
↓ 1 callersMethodset_attention_slice
(self, slice_size: Optional[int])
DeepCache/svd/pipeline_utils.py:2098
↓ 1 callersMethodset_attention_slice
r""" Enable sliced attention computation. When this option is enabled, the attention module splits the input tensor in slices to comp
DeepCache/sd/unet_2d_condition.py:670
↓ 1 callersMethodset_attention_slice
(self, slice_size: Optional[int])
DeepCache/sd/pipeline_utils.py:1831
↓ 1 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
DeepCache/sdxl/unet_2d_condition.py:619
↓ 1 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
DeepCache/svd/unet_spatio_temporal_condition.py:276
↓ 1 callersMethodset_interval
(self, total_number, replicate_interval=None, nonuniform=False, pow=None)
experiments/ldm/ldm/models/diffusion/ddpm.py:1414
↓ 1 callersMethodsetup_image_folder
(self, image_folder)
experiments/ddpm/ddpm/utils/logging.py:73
↓ 1 callersMethodsetup_sublogger
(self, sub_name, sub_config, append=False)
experiments/ddpm/ddpm/utils/logging.py:24
↓ 1 callersFunctionsigmoid
(x)
experiments/ddpm/ddpm/runners/diffusion.py:29
↓ 1 callersFunctionsigmoid
(x)
experiments/ddpm/ddpm/runners/deepcache.py:31
↓ 1 callersFunctionsynset2idx
(path_to_yaml="data/index_synset.yaml")
experiments/ldm/ldm/data/imagenet.py:20
↓ 1 callersFunctiontensor2vid
(video: torch.Tensor, processor, output_type="np")
DeepCache/svd/pipeline_stable_video_diffusion.py:42
↓ 1 callersMethodtrain
(self)
experiments/ddpm/ddpm/runners/diffusion.py:96
↓ 1 callersFunctiontrain_searcher
(opt, metric='dot_product', partioning_trainsize=None,
experiments/ldm/scripts/train_searcher.py:62
↓ 1 callersMethodtrain_searcher
(self, k, metric='dot_product', searcher_savedir=None)
experiments/ldm/scripts/knn2img.py:75
↓ 1 callersMethodunwrap_modules
(self)
DeepCache/extension/deepcache.py:83
↓ 1 callersMethodupcast_vae
(self)
DeepCache/sdxl/pipeline_stable_diffusion_xl.py:536
↓ 1 callersMethodupcast_vae
(self)
DeepCache/sdxl/pipeline_stable_diffusion_xl_img2img.py:645
↓ 1 callersFunctionvisualize_cond_img
(path)
experiments/ldm/notebook_helpers.py:127
↓ 1 callersFunctionwarn_deprecated_model_variant
(pretrained_model_name_or_path, use_auth_token, variant, revision, model_filenames)
DeepCache/sdxl/pipeline_utils.py:254
↓ 1 callersFunctionwarn_deprecated_model_variant
(pretrained_model_name_or_path, use_auth_token, variant, revision, model_filenames)
DeepCache/svd/pipeline_utils.py:252
↓ 1 callersFunctionwarn_deprecated_model_variant
(pretrained_model_name_or_path, use_auth_token, variant, revision, model_filenames)
DeepCache/sd/pipeline_utils.py:254
↓ 1 callersFunctionwarp_single_latent
Warp latent of a single frame with given flow Args: latent: latent code of a single frame reference_flow: flow which to warp
DeepCache/sd/pipeline_text_to_video_zero.py:238
↓ 1 callersMethodwrap_modules
(self)
DeepCache/extension/deepcache.py:57
↓ 1 callersMethodwrap_unet_forward
(self)
DeepCache/extension/deepcache.py:38
↓ 1 callersMethodwrite_description_to_folder
(self, file_name, config)
experiments/ddpm/ddpm/utils/logging.py:103
↓ 1 callersMethodwrite_logs
(self, loss, logits, targets)
experiments/ldm/ldm/models/diffusion/classifier.py:162
↓ 1 callersFunctionzero_module
Zero out the parameters of a module and return it.
experiments/ldm/ldm/modules/attention.py:67
Method__call__
r""" Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*):
DeepCache/sdxl/pipeline_stable_diffusion_xl.py:557
Method__call__
r""" Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*):
DeepCache/sdxl/pipeline_stable_diffusion_xl_img2img.py:666
Method__call__
r""" The call function to the pipeline for generation. Args: image (`PIL.Image.Image` or `List[PIL.Image.Image]` or `torc
DeepCache/svd/pipeline_stable_video_diffusion.py:299
Method__call__
r""" The call function to the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*): T
DeepCache/sd/pipeline_stable_diffusion.py:564
Method__call__
(self, attn, hidden_states, encoder_hidden_states=None, attention_mask=None)
DeepCache/sd/pipeline_text_to_video_zero.py:92
Method__call__
(self, attn, hidden_states, encoder_hidden_states=None, attention_mask=None)
DeepCache/sd/pipeline_text_to_video_zero.py:153
Method__call__
The call function to the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*): The p
DeepCache/sd/pipeline_text_to_video_zero.py:476
Method__call__
(self, input, target)
experiments/ddpm/ddpm/datasets/vision.py:63
Method__call__
(self, img)
experiments/ddpm/ddpm/datasets/__init__.py:21
Method__call__
(self, x, n)
experiments/ldm/scripts/knn2img.py:163
Method__call__
(self, n, **kwargs)
experiments/ldm/ldm/lr_scheduler.py:32
Method__call__
(self, n, **kwargs)
experiments/ldm/ldm/lr_scheduler.py:77
Method__getitem__
(self, i)
experiments/ddpm/fid.py:92
Method__getitem__
(self, index)
experiments/ddpm/ddpm/datasets/ffhq.py:31
Method__getitem__
(self, index)
experiments/ddpm/ddpm/datasets/vision.py:28
Method__getitem__
(self, index)
experiments/ddpm/ddpm/datasets/lsun.py:38
Method__getitem__
Args: index (int): Index Returns: tuple: Tuple (image, target) where target is the index of the target categ
experiments/ddpm/ddpm/datasets/lsun.py:146
Method__getitem__
(self, index)
experiments/ddpm/ddpm/datasets/celeba.py:133
Method__getitem__
(self, idx)
experiments/ldm/main.py:142
Method__getitem__
(self, i)
experiments/ldm/ldm/data/imagenet.py:42
Method__getitem__
(self, i)
experiments/ldm/ldm/data/imagenet.py:339
Method__getitem__
(self, i)
experiments/ldm/ldm/data/lsun.py:39
Method__init__
( self, sample_size: Optional[int] = None, in_channels: int = 4, out_channels:
DeepCache/sdxl/unet_2d_condition.py:164
Method__init__
(self, in_channels: int, out_channels: int, act_fn: str)
DeepCache/sdxl/unet_2d_blocks.py:461
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
DeepCache/sdxl/unet_2d_blocks.py:483
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
DeepCache/sdxl/unet_2d_blocks.py:574
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
DeepCache/sdxl/unet_2d_blocks.py:716
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:840
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:957
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:1118
Method__init__
( self, in_channels: int, out_channels: int, dropout: float = 0.0, num
DeepCache/sdxl/unet_2d_blocks.py:1208
Method__init__
( self, in_channels: int, out_channels: int, dropout: float = 0.0, num
DeepCache/sdxl/unet_2d_blocks.py:1268
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:1353
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:1455
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:1534
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:1632
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
DeepCache/sdxl/unet_2d_blocks.py:1792
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, cross
DeepCache/sdxl/unet_2d_blocks.py:1870
← previousnext →401–500 of 1,211, ranked by callers