MCPcopy Create free account

hub / github.com/bmaltais/dehydrate / functions

Functions30 in github.com/bmaltais/dehydrate

⨍ Functions 30◇ Types & classes 0
↓ 13 callersFunctionassign_to_checkpoint
This does the final conversion step: take locally converted weights and apply a global renaming to them. It splits attention layers, and take
convert_ckpt_diffusers.py:143
↓ 6 callersFunction_copy_linear
(hf_linear, pt_linear)
convert_ckpt_diffusers.py:557
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
convert_ckpt_diffusers.py:54
↓ 5 callersFunctionshave_segments
Removes segments. Positive values shave the first segments, negative shave the last segments.
convert_ckpt_diffusers.py:44
↓ 4 callersFunctionrenew_vae_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
convert_ckpt_diffusers.py:76
↓ 3 callersFunctionrenew_attention_paths
Updates paths inside attentions to the new naming scheme (local renaming)
convert_ckpt_diffusers.py:92
↓ 2 callersFunctionconv_attn_to_linear
(checkpoint)
convert_ckpt_diffusers.py:195
↓ 2 callersFunctionrenew_vae_attention_paths
Updates paths inside attentions to the new naming scheme (local renaming)
convert_ckpt_diffusers.py:113
↓ 1 callersFunction_copy_attn_layer
(hf_attn_layer, pt_attn_layer)
convert_ckpt_diffusers.py:549
↓ 1 callersFunction_copy_layer
(hf_layer, pt_layer)
convert_ckpt_diffusers.py:561
↓ 1 callersFunction_copy_layers
(hf_layers, pt_layers)
convert_ckpt_diffusers.py:574
↓ 1 callersFunctioncompressed_pickle
(title, data)
ckpt_subtract.py:18
↓ 1 callersFunctionconvert_diffusers_vae
(vae_path)
ckpt_swap_vae.py:91
↓ 1 callersFunctionconvert_ldm_bert_checkpoint
(checkpoint, config)
convert_ckpt_diffusers.py:548
↓ 1 callersFunctionconvert_ldm_clip_checkpoint
(checkpoint)
convert_ckpt_diffusers.py:598
↓ 1 callersFunctionconvert_ldm_unet_checkpoint
Takes a state dict and a config, and returns a converted checkpoint.
convert_ckpt_diffusers.py:288
↓ 1 callersFunctionconvert_ldm_vae_checkpoint
(checkpoint, config)
convert_ckpt_diffusers.py:441
↓ 1 callersFunctionconvert_text_enc_state_dict
(text_enc_dict)
convert_diffusers_to_ckpt.py:195
↓ 1 callersFunctionconvert_unet_state_dict
(unet_state_dict)
convert_diffusers_to_ckpt.py:92
↓ 1 callersFunctionconvert_vae_state_dict
(vae_state_dict)
convert_diffusers_to_ckpt.py:168
↓ 1 callersFunctionconvert_vae_state_dict
(vae_state_dict)
ckpt_swap_vae.py:70
↓ 1 callersFunctioncreate_ldm_bert_config
(original_config)
convert_ckpt_diffusers.py:278
↓ 1 callersFunctioncreate_unet_diffusers_config
Creates a config for the diffusers based on the config of the LDM model.
convert_ckpt_diffusers.py:207
↓ 1 callersFunctioncreate_vae_diffusers_config
Creates a config for the diffusers based on the config of the LDM model.
convert_ckpt_diffusers.py:244
↓ 1 callersFunctiondecompress_pickle
(file)
uncompress.py:16
↓ 1 callersFunctiondecompress_pickle
(file)
ckpt_add.py:18
↓ 1 callersFunctionmerge_vae
(ckpt, vae, output)
ckpt_swap_vae.py:97
↓ 1 callersFunctionreshape_weight_for_sd
(w)
convert_diffusers_to_ckpt.py:163
↓ 1 callersFunctionreshape_weight_for_sd
(w)
ckpt_swap_vae.py:65
Functioncreate_diffusers_schedular
(original_config)
convert_ckpt_diffusers.py:268