MCPcopy Create free account

hub / github.com/alibaba/animate-anything / functions

Functions387 in github.com/alibaba/animate-anything

↓ 9 callersMethodchunk
(self, it, size)
utils/dataset.py:293
↓ 7 callersFunctioncalculate_latent_motion_score
(latents)
utils/common.py:296
↓ 7 callersFunctioncustom_checkpoint
(module, mode=None)
models/unet_3d_blocks.py:32
↓ 7 callersMethodtrain
(self, mode: bool = True)
stable_lora/lora.py:106
↓ 6 callersMethod__call__
(self, x, latent)
models/layerdiffuse_VAE.py:176
↓ 6 callersFunctionget_train_dataset
(dataset_types, train_data, tokenizer)
utils/dataset.py:626
↓ 5 callersFunctionDDPM_forward_timesteps
larger step -> smaller t -> smaller alphas[t:] -> smaller xt -> smaller x0
utils/common.py:32
↓ 5 callersMethod__init__
( self, in_channels, out_channels, kernel_size, ratio_gin, rat
utils/lama.py:119
↓ 5 callersFunctionget_prompt_ids
(prompt, tokenizer)
utils/dataset.py:44
↓ 5 callersMethodis_stable_lora
(self)
utils/lora_handler.py:99
↓ 5 callersFunctionnormalize_input
( item, mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225], use_simple_norm=True )
utils/dataset.py:23
↓ 5 callersFunctionsensible_buckets
(m_width, m_height, w, h, min_size=192)
utils/bucketing.py:22
↓ 5 callersFunctiontensor_to_vae_latent
(t, vae)
utils/common.py:12
↓ 4 callersMethod__init__
( self, in_channels: int, prev_output_channel: int, out_channels: int,
models/unet_3d_blocks.py:769
↓ 4 callersMethod__len__
(self)
utils/dataset.py:427
↓ 4 callersFunction_mean_pixel_distance
Return the mean average distance in pixel values between `left` and `right`. Both `left and `right` should be 2 dimensional 8-bit images of the sa
utils/common.py:287
↓ 4 callersFunctionget_frame_batch
(max_frames, sample_fps, vr, transform)
utils/dataset.py:92
↓ 4 callersFunctionget_moved_area_mask
(frames, move_th=5, th=-1)
utils/common.py:88
↓ 4 callersMethodis_cloneofsimo_lora
(self)
utils/lora_handler.py:96
↓ 4 callersFunctionparam_optim
(model, condition, extra_params=None, is_lora=False, negation=None)
train_lora.py:161
↓ 3 callersFunction_text_lora_path
(path: str)
utils/lora.py:1086
↓ 3 callersFunctioncalculate_motion_precision
(frames, mask)
utils/common.py:136
↓ 3 callersFunctioncalculate_motion_score
(frame_imgs, calculate_edges=False, color="RGB")
utils/common.py:307
↓ 3 callersFunctioncast_to_gpu_and_type
(model_list, device, weight_dtype)
train_lora.py:251
↓ 3 callersFunctioncreate_optim_params
(name='param', params=None, lr=5e-6, extra_params=None)
train_lora.py:172
↓ 3 callersFunctioncreate_optim_params
(name='param', params=None, lr=5e-6, extra_params=None)
train_transparent_i2v_stage2.py:195
↓ 3 callersFunctioncross_attn_g_c
( attn, temp_attn, resnet, temp_conv, hidden_states, enco
models/unet_3d_blocks.py:72
↓ 3 callersMethoddeactivate_lora_train
Usage: Use before and after sampling previews. Currently only available for Stable LoRA.
utils/lora_handler.py:270
↓ 3 callersFunctionextend_datasets
(datasets, dataset_items, extend=False)
utils/dataset.py:639
↓ 3 callersFunctionextract_lora_ups_down
(model, target_replace_module=DEFAULT_TARGET_REPLACE)
utils/lora.py:528
↓ 3 callersFunctionfilter_dict
(_dict, keys=[])
utils/lora_handler.py:59
↓ 3 callersMethodget_empty_store
()
utils/ptp_utils.py:217
↓ 3 callersFunctionmonkeypatch_or_replace_lora
( model, loras, target_replace_module=DEFAULT_TARGET_REPLACE, r: Union[int, List[int]] = 4, )
utils/lora.py:817
↓ 3 callersFunctionparse_safeloras
Converts a loaded safetensor file that contains a set of module Loras into Parameters and other information Output is a dictionary of {
utils/lora.py:683
↓ 3 callersFunctionparse_safeloras_embeds
Converts a loaded safetensor file that contains Textual Inversion embeds into a dictionary of embed_token: Tensor
utils/lora.py:744
↓ 3 callersFunctionsave_lora
( unet=None, text_encoder=None, save_text_weights=False, output_dir="output"
stable_lora/lora.py:304
↓ 3 callersFunctionsave_lora_weight
( model, path="./lora.pt", target_replace_module=DEFAULT_TARGET_REPLACE, )
utils/lora.py:569
↓ 3 callersFunctionup_down_g_c
(resnet, temp_conv, hidden_states, temb, num_frames)
models/unet_3d_blocks.py:115
↓ 2 callersMethod__getname__
()
utils/dataset.py:425
↓ 2 callersMethod__init__
( self, in_channels: int, out_channels: int, kernel_size: int, r: in
stable_lora/lora.py:71
↓ 2 callersMethod__init__
( self, in_features, out_features, bias=False, r=4, dropout_p=0.1, scale=1.0 )
utils/lora.py:34
↓ 2 callersFunction_append_dims
Appends dimensions to the end of a tensor until it has target_dims dimensions.
models/pipeline.py:216
↓ 2 callersFunction_ti_lora_path
(path: str)
utils/lora.py:1091
↓ 2 callersMethodadd_lora_to_model
(self, use_lora, model, replace_modules, dropout=0.0, lora_path='', r=16)
utils/lora_handler.py:239
↓ 2 callersFunctionapply_learned_embed_in_clip
( learned_embeds, text_encoder, tokenizer, token: Optional[Union[str, List[str]]] = None,
utils/lora.py:1096
↓ 2 callersFunctionbatch_eval
(unet, text_encoder, vae, vae_processor, pretrained_model_path, validation_data, output_dir, preview, glo
train_lora.py:892
↓ 2 callersFunctionbatch_eval
(unet, text_encoder, vae, vae_processor, pretrained_model_path, validation_data, output_dir, preview, glo
train.py:793
↓ 2 callersFunctioncast_to_gpu_and_type
(model_list, device, weight_dtype)
train.py:228
↓ 2 callersMethodcheck_lora_ext
(self, lora_file: str)
utils/lora_handler.py:122
↓ 2 callersFunctionclosest_bucket
(m_size, size, direction, min_size)
utils/bucketing.py:16
↓ 2 callersFunctioncollapse_lora
(model, alpha=1.0)
utils/lora.py:780
↓ 2 callersFunctioneval
(pipeline, vae_processor, validation_data, out_file, index, forward_t=25, preview=True)
train_svd.py:726
↓ 2 callersFunctionfreeze_models
(models_to_freeze)
train_lora.py:120
↓ 2 callersFunctionfreeze_models
(models_to_freeze)
train.py:117
↓ 2 callersFunctionget_down_block
( down_block_type, num_layers, in_channels, out_channels, temb_channels, add_downsampl
models/unet_3d_blocks.py:122
↓ 2 callersMethodget_lora_file_path
( self, lora_path: str, model: Union[UNet3DConditionModel, CLIPTextModel] )
utils/lora_handler.py:125
↓ 2 callersMethodget_lora_func
(self, func_type: LORA_FUNC_TYPES = LoraFuncTypes.loader)
utils/lora_handler.py:102
↓ 2 callersFunctionget_up_block
( up_block_type, num_layers, in_channels, out_channels, prev_output_channel, temb_chan
models/unet_3d_blocks.py:178
↓ 2 callersFunctionget_word_inds
(text: str, word_place: int, tokenizer)
utils/seq_aligner.py:131
↓ 2 callersFunctionhandle_memory_attention
(enable_xformers_memory_efficient_attention, enable_torch_2_attn, unet)
train_lora.py:143
↓ 2 callersFunctionhandle_memory_attention
(enable_xformers_memory_efficient_attention, enable_torch_2_attn, unet)
train.py:140
↓ 2 callersFunctionload_primary_models
(pretrained_model_path, in_channels=-1, motion_strength=False)
train_lora.py:88
↓ 2 callersFunctionload_primary_models
(pretrained_model_path, in_channels=-1, motion_strength=False)
train.py:85
↓ 2 callersFunctionload_primary_models
(pretrained_model_path, eval=False)
train_svd.py:85
↓ 2 callersFunctionmonkeypatch_or_replace_lora_extended
( model, loras, target_replace_module=DEFAULT_TARGET_REPLACE, r: Union[int, List[int]] = 4, )
utils/lora.py:861
↓ 2 callersFunctionmonkeypatch_remove_lora
(model)
utils/lora.py:995
↓ 2 callersFunctionordered_g_c
(idx)
models/unet_3d_blocks.py:85
↓ 2 callersFunctionprocess_video
(vid_path, use_bucketing, w, h, get_frame_buckets, get_frame_batch)
utils/dataset.py:108
↓ 2 callersMethodprocess_video_wrapper
(self, vid_path)
utils/dataset.py:497
↓ 2 callersFunctionread_video
(video_path, frame_number=-1)
utils/common.py:64
↓ 2 callersMethodreset_parameters
(self)
stable_lora/lora.py:99
↓ 2 callersFunctionresolve_bucket
(i,h,w)
utils/bucketing.py:20
↓ 2 callersFunctionsafe_save
( tensors: Dict[str, torch.Tensor], filename: str, metadata: Optional[Dict[str, str]]
utils/lora.py:21
↓ 2 callersFunctionsave_pipe
( path, global_step, accelerator, unet, text_encoder, vae,
train_lora.py:292
↓ 2 callersFunctionsave_pipe
( path, global_step, accelerator, unet, text_encoder, vae,
train.py:269
↓ 2 callersFunctionsave_pipe
( path, global_step, accelerator, unet, text_encoder, vae,
train_svd.py:298
↓ 2 callersFunctionsave_safeloras_with_embeds
Saves the Lora from multiple modules in a single safetensor file. modelmap is a dictionary of { "module name": (module, target_repla
utils/lora.py:596
↓ 2 callersFunctionupdate_alpha_time_word
(alpha, bounds: Union[float, Tuple[float, float]], prompt_ind: int, word_inds: Opti
utils/ptp_utils.py:384
↓ 2 callersFunctionzero_module
(module)
models/layerdiffuse_VAE.py:11
↓ 1 callersFunctionDDPM_forward
(x0, step, num_frames, scheduler)
utils/common.py:22
↓ 1 callersMethod__getname__
()
utils/dataset.py:233
↓ 1 callersMethod__getname__
()
utils/dataset.py:339
↓ 1 callersMethod__getname__
()
utils/dataset.py:509
↓ 1 callersMethod__getname__
()
utils/dataset.py:577
↓ 1 callersMethod__init__
(self)
utils/ptp_utils.py:246
↓ 1 callersMethod__init__
( self, in_channels: int = 3, out_channels: int = 4, down_block_types: Tuple[s
models/layerdiffuse_VAE.py:46
↓ 1 callersFunction_detect_edges
Detect edges using the luma channel of a frame. Arguments: lum: 2D 8-bit image representing the luma channel of a frame. Returns:
utils/common.py:259
↓ 1 callersMethod_encode_prompt
r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):
models/pipeline_stage2.py:12
↓ 1 callersFunctionaccelerate_set_verbose
(accelerator)
train_lora.py:70
↓ 1 callersFunctionaccelerate_set_verbose
(accelerator)
train.py:67
↓ 1 callersFunctionaccelerate_set_verbose
(accelerator)
train_svd.py:67
↓ 1 callersFunctionactivate_lora_train
(model, bias)
stable_lora/lora.py:250
↓ 1 callersFunctionbatch_eval
(unet, text_encoder, vae, vae_alpha_encoder, vae_alpha_decoder, vae_processor, lora_manager, pretrained_model_
train_transparent_i2v_stage2.py:555
↓ 1 callersMethodbetween_steps
(self)
utils/ptp_utils.py:181
↓ 1 callersMethodbuild_json
(self, json_data)
utils/dataset.py:158
↓ 1 callersMethodbuild_json_dict
(self, data, nested_data, extended_data)
utils/dataset.py:170
↓ 1 callersFunctioncast_to_gpu_and_type
(model_list, device, weight_dtype)
train_transparent_i2v_stage2.py:274
↓ 1 callersFunctioncast_to_gpu_and_type
(model_list, device, weight_dtype)
train_svd.py:226
↓ 1 callersFunctionconvert_loras_to_safeloras_with_embeds
Converts the Lora from multiple pytorch .pt files into a single safetensor file. modelmap is a dictionary of { "module name": (pytor
utils/lora.py:638
↓ 1 callersFunctionconvert_text_enc_state_dict
(text_enc_dict)
utils/convert_diffusers_to_original_ms_text_to_video.py:298
next →1–100 of 387, ranked by callers