Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/horseee/DeepCache
/ functions
Functions
1,211 in github.com/horseee/DeepCache
⨍
Functions
1,211
◇
Types & classes
220
Method
_val_dataloader
(self, shuffle=False)
experiments/ldm/main.py:207
Function
accumulate_flops
(self, layer_flops)
DeepCache/flops.py:378
Function
add_Poisson_noise
(img)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:408
Function
add_flops_counter_hook_function
(module, ost, verbose, ignore_list)
DeepCache/flops.py:444
Function
add_flops_counter_variable_or_reset
(module)
DeepCache/flops.py:524
Function
add_resize
(img, sf=4)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:343
Function
add_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
Function
add_speckle_noise
(img, noise_level1=2, noise_level2=25)
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:390
Function
analytic_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
Function
analytic_kernel
Calculate the X4 kernel from the X2 kernel (for proof see appendix in paper)
experiments/ldm/ldm/modules/image_degradation/bsrgan.py:49
Method
attn_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
Method
attn_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
Method
attn_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
Function
augment_img_np3
(img, mode=0)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:441
Function
augment_img_tensor
Kai Zhang (github: https://github.com/cszn)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:422
Function
augment_img_tensor4
Kai Zhang (github: https://github.com/cszn)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:401
Function
augment_imgs
(img_list, hflip=True, rot=True)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:469
Function
bar_update
(count, block_size, total_size)
experiments/ddpm/ddpm/datasets/utils.py:11
Function
batch_counter_hook
(module, input, output)
DeepCache/flops.py:492
Function
betas_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
Function
blur
x: image, NxcxHxW k: kernel, Nx1xhxw
experiments/ldm/ldm/modules/image_degradation/bsrgan_light.py:128
Function
blur
x: image, NxcxHxW k: kernel, Nx1xhxw
experiments/ldm/ldm/modules/image_degradation/bsrgan.py:128
Function
bn_flops_counter_hook
(module, input, output)
DeepCache/flops.py:84
Function
calculate_psnr
(img1, img2, border=0)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:621
Function
calculate_ssim
calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255]
experiments/ldm/ldm/modules/image_degradation/utils_image.py:642
Function
channel_convert
(in_c, tar_type, img_list)
experiments/ldm/ldm/modules/image_degradation/utils_image.py:597
Function
classical_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
Function
classical_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
Method
components
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
Method
components
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
Method
components
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
Function
compute_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
Method
configure_optimizers
(self)
experiments/ldm/ldm/models/autoencoder.py:197
Method
configure_optimizers
(self)
experiments/ldm/ldm/models/autoencoder.py:386
Method
configure_optimizers
(self)
experiments/ldm/ldm/models/diffusion/classifier.py:220
Method
configure_optimizers
(self)
experiments/ldm/ldm/models/diffusion/ddpm.py:415
Method
configure_optimizers
(self)
experiments/ldm/ldm/models/diffusion/ddpm.py:1361
Function
conv_flops_counter_hook
(conv_module, input, output)
DeepCache/flops.py:99
Function
convert_module_to_f16
(x)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:24
Function
convert_module_to_f32
(x)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:27
Method
convert_to_fp16
Convert the torso of the model to float16.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:694
Method
convert_to_fp16
Convert the torso of the model to float16.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:943
Method
convert_to_fp32
Convert the torso of the model to float32.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:702
Method
convert_to_fp32
Convert the torso of the model to float32.
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:950
Method
count_flops
(model, _x, y)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:375
Method
count_flops
(model, _x, y)
experiments/ldm/ldm/modules/diffusionmodules/openaimodel.py:409
Function
count_ops_and_params
(model, example_inputs, layer_wise=False)
DeepCache/flops.py:20
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:677
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:1066
Method
create_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:1176
Method
create_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:1603
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:1754
Method
create_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:1843
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:1949
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:2221
Method
create_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:2327
Method
create_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:2778
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:2933
Method
create_custom_forward
(module)
DeepCache/sdxl/unet_2d_blocks.py:3022
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sdxl/unet_2d_blocks.py:3149
Method
create_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:1017
Method
create_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1190
Method
create_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1394
Method
create_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:1549
Method
create_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1708
Method
create_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:1940
Method
create_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:2031
Method
create_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:2148
Method
create_custom_forward
(module)
DeepCache/svd/unet_3d_blocks.py:2257
Method
create_custom_forward
(module, return_dict=None)
DeepCache/svd/unet_3d_blocks.py:2372
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:677
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:1066
Method
create_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:1176
Method
create_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:1601
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:1752
Method
create_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:1841
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:1947
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:2219
Method
create_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:2319
Method
create_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:2771
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:2926
Method
create_custom_forward
(module)
DeepCache/sd/unet_2d_blocks.py:3015
Method
create_custom_forward
(module, return_dict=None)
DeepCache/sd/unet_2d_blocks.py:3142
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:678
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1067
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1177
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1604
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1755
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1844
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:1950
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2222
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2328
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2779
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:2934
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:3023
Method
custom_forward
(*inputs)
DeepCache/sdxl/unet_2d_blocks.py:3150
Method
custom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1018
Method
custom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1191
Method
custom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1395
Method
custom_forward
(*inputs)
DeepCache/svd/unet_3d_blocks.py:1550
← previous
next →
701–800 of 1,211, ranked by callers