MCPcopy Create free account

hub / github.com/horseee/DeepCache / functions

Functions1,211 in github.com/horseee/DeepCache

Method_val_dataloader
(self, shuffle=False)
experiments/ldm/main.py:207
Functionaccumulate_flops
(self, layer_flops)
DeepCache/flops.py:378
Functionadd_Poisson_noise
(img)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:408
Functionadd_flops_counter_hook_function
(module, ost, verbose, ignore_list)
DeepCache/flops.py:444
Functionadd_flops_counter_variable_or_reset
(module)
DeepCache/flops.py:524
Functionadd_resize
(img, sf=4)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:343
Functionadd_sharpening
USM sharpening. borrowed from real-ESRGAN Input image: I; Blurry image: B. 1. K = I + weight * (I - B) 2. Mask = 1 if abs(I - B) > thresho
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:299
Functionadd_speckle_noise
(img, noise_level1=2, noise_level2=25)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:390
Functionanalytic_kernel
Calculate the X4 kernel from the X2 kernel (for proof see appendix in paper)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:49
Functionanalytic_kernel
Calculate the X4 kernel from the X2 kernel (for proof see appendix in paper)
experiments/ldm/ldm/modules/image_degradation/bsrgan.py:49
Methodattn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
DeepCache/sdxl/unet_2d_condition.py:596
Methodattn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
DeepCache/svd/unet_spatio_temporal_condition.py:249
Methodattn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
DeepCache/sd/unet_2d_condition.py:596
Functionaugment_img_np3
(img, mode=0)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:441
Functionaugment_img_tensor
Kai Zhang (github: https://github.com/cszn)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:422
Functionaugment_img_tensor4
Kai Zhang (github: https://github.com/cszn)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:401
Functionaugment_imgs
(img_list, hflip=True, rot=True)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:469
Functionbar_update
(count, block_size, total_size)
experiments/ddpm/ddpm/datasets/utils.py:11
Functionbatch_counter_hook
(module, input, output)
DeepCache/flops.py:492
Functionbetas_for_alpha_bar
Create a beta schedule that discretizes the given alpha_t_bar function, which defines the cumulative product of (1-beta) over time from t = [
experiments/ldm/ldm/modules/diffusionmodules/util.py:77
Functionblur
x: image, NxcxHxW k: kernel, Nx1xhxw
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:128
Functionblur
x: image, NxcxHxW k: kernel, Nx1xhxw
experiments/ldm/ldm/modules/image_degradation/bsrgan.py:128
Functionbn_flops_counter_hook
(module, input, output)
DeepCache/flops.py:84
Functioncalculate_psnr
(img1, img2, border=0)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:621
Functioncalculate_ssim
calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255]
experiments/ldm/ldm/modules/image_degradation/utils_image.py:642
Functionchannel_convert
(in_c, tar_type, img_list)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:597
Functionclassical_degradation
blur + downsampling Args: x: HxWxC image, [0, 1]/[0, 255] k: hxw, double sf: down-scale factor Return: downsa
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:284
Functionclassical_degradation
blur + downsampling Args: x: HxWxC image, [0, 1]/[0, 255] k: hxw, double sf: down-scale factor Return: downsa
experiments/ldm/ldm/modules/image_degradation/bsrgan.py:284
Methodcomponents
r""" The `self.components` property can be useful to run different pipelines with the same weights and configurations without realloca
DeepCache/sdxl/pipeline_utils.py:1662
Methodcomponents
r""" The `self.components` property can be useful to run different pipelines with the same weights and configurations without realloca
DeepCache/svd/pipeline_utils.py:1929
Methodcomponents
r""" The `self.components` property can be useful to run different pipelines with the same weights and configurations without realloca
DeepCache/sd/pipeline_utils.py:1662
Functioncompute_average_flops_cost
A method that will be available after add_flops_counting_methods() is called on a desired net object. Returns current mean flops consumpt
DeepCache/flops.py:408
Methodconfigure_optimizers
(self)
experiments/ldm/ldm/models/autoencoder.py:197
Methodconfigure_optimizers
(self)
experiments/ldm/ldm/models/autoencoder.py:386
Methodconfigure_optimizers
(self)
experiments/ldm/ldm/models/diffusion/classifier.py:220
Methodconfigure_optimizers
(self)
experiments/ldm/ldm/models/diffusion/ddpm.py:415
Methodconfigure_optimizers
(self)
experiments/ldm/ldm/models/diffusion/ddpm.py:1361
Functionconv_flops_counter_hook
(conv_module, input, output)
DeepCache/flops.py:99
Functionconvert_module_to_f16
(x)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:24
Functionconvert_module_to_f32
(x)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:27
Methodconvert_to_fp16
Convert the torso of the model to float16.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:694
Methodconvert_to_fp16
Convert the torso of the model to float16.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:943
Methodconvert_to_fp32
Convert the torso of the model to float32.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:702
Methodconvert_to_fp32
Convert the torso of the model to float32.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:950
Methodcount_flops
(model, _x, y)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:375
Methodcount_flops
(model, _x, y)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:409
Functioncount_ops_and_params
(model, example_inputs, layer_wise=False)
DeepCache/flops.py:20
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:677
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:1066
Methodcreate_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:1176
Methodcreate_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:1603
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:1754
Methodcreate_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:1843
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:1949
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:2221
Methodcreate_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:2327
Methodcreate_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:2778
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:2933
Methodcreate_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:3022
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:3149
Methodcreate_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:1017
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1190
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1394
Methodcreate_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:1549
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1708
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1940
Methodcreate_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:2031
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:2148
Methodcreate_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:2257
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:2372
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:677
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:1066
Methodcreate_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:1176
Methodcreate_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:1601
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:1752
Methodcreate_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:1841
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:1947
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:2219
Methodcreate_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:2319
Methodcreate_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:2771
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:2926
Methodcreate_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:3015
Methodcreate_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:3142
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:678
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1067
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1177
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1604
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1755
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1844
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1950
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2222
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2328
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2779
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2934
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:3023
Methodcustom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:3150
Methodcustom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1018
Methodcustom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1191
Methodcustom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1395
Methodcustom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1550
← previousnext →701–800 of 1,211, ranked by callers