MCPcopy Create free account

hub / github.com/Omni-Avatar/OmniAvatar / functions

Functions234 in github.com/Omni-Avatar/OmniAvatar

⨍ Functions 234◇ Types & classes 48
↓ 67 callersMethodto
(self, device)
OmniAvatar/models/model_manager.py:391
↓ 11 callersFunctionload_state_dict
(file_path, torch_dtype=None)
OmniAvatar/utils/io_utils.py:77
↓ 10 callersFunctionhash_state_dict_keys
(state_dict, with_shape=True)
OmniAvatar/utils/io_utils.py:182
↓ 8 callersMethod__init__
(self, dim, channel_first=True, images=True, bias=False)
OmniAvatar/models/wan_video_vae.py:57
↓ 8 callersMethod__init__
(self, in_dim, out_dim)
OmniAvatar/models/wan_video_dit.py:229
↓ 8 callersMethodupdate
(self, hidden_states)
OmniAvatar/wan_video.py:340
↓ 7 callersMethodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, **kwargs)
OmniAvatar/models/model_manager.py:105
↓ 6 callersFunctioncheck_is_instance
(model, module_class)
OmniAvatar/models/wan_video_vae.py:11
↓ 6 callersMethodload_models_to_device
(self, loadmodel_names=[])
OmniAvatar/base.py:95
↓ 5 callersMethod__init__
(self, dim, eps=1e-6)
OmniAvatar/models/wan_video_text_encoder.py:24
↓ 5 callersMethodmatch
(self, file_path="", state_dict={})
OmniAvatar/models/model_manager.py:102
↓ 4 callersFunctioncreate_custom_forward
(module)
OmniAvatar/distributed/xdit_context_parallel.py:74
↓ 4 callersMethoddecode
(self, z, scale)
OmniAvatar/models/wan_video_vae.py:552
↓ 4 callersFunctionenable_vram_management
(model: torch.nn.Module, module_map: dict, module_config: dict, max_num_param=None, overflow_module_config: di
OmniAvatar/vram_management/layers.py:92
↓ 4 callersMethodencode
(self, x, scale)
OmniAvatar/models/wan_video_vae.py:525
↓ 4 callersMethodencode
Encodes the input features into the output space. Args: extract_features (torch.Tensor): The extracted features from the
OmniAvatar/models/wav2vec.py:133
↓ 4 callersMethodfetch_model
(self, model_name, file_path=None, require_model_path=False)
OmniAvatar/models/model_manager.py:369
↓ 4 callersMethodnorm
(self, x)
OmniAvatar/models/wan_video_dit.py:108
↓ 3 callersFunctionbasic_clean
(text)
OmniAvatar/prompters/wan_prompter.py:11
↓ 3 callersMethodfrom_model_manager
(model_manager: ModelManager, torch_dtype=None, device=None, use_usp=False, infer=False)
OmniAvatar/wan_video.py:138
↓ 3 callersFunctionload_model_from_single_file
(state_dict, model_names, model_classes, model_resource, torch_dtype, device, infer)
OmniAvatar/models/model_manager.py:8
↓ 3 callersMethodpatchify
(self, x: torch.Tensor)
OmniAvatar/models/wan_video_dit.py:320
↓ 3 callersFunctionprecompute_freqs_cis
(dim: int, end: int = 1024, theta: float = 10000.0)
OmniAvatar/models/wan_video_dit.py:85
↓ 3 callersMethodpreprocess_image
(self, image)
OmniAvatar/base.py:30
↓ 2 callersFunctionadd_silence_to_audio_ffmpeg
(audio_path, tmp_audio_path, silence_duration_s=0.5)
OmniAvatar/utils/audio_preprocess.py:4
↓ 2 callersMethodbuild_1d_mask
(self, length, left_bound, right_bound, border_width)
OmniAvatar/models/wan_video_vae.py:621
↓ 2 callersMethodbuild_mask
(self, data, is_bound, border_width)
OmniAvatar/models/wan_video_vae.py:630
↓ 2 callersFunctioncast_to
(weight, dtype, device)
OmniAvatar/vram_management/layers.py:5
↓ 2 callersMethodclear_cache
(self)
OmniAvatar/models/wan_video_vae.py:589
↓ 2 callersFunctionconvert_state_dict_keys_to_single_str
(state_dict, with_shape=True)
OmniAvatar/utils/io_utils.py:232
↓ 2 callersFunctioncount_conv3d
(model)
OmniAvatar/models/wan_video_vae.py:484
↓ 2 callersMethoddecode_video
(self, latents, tiled=True, tile_size=(34, 34), tile_stride=(18, 16))
OmniAvatar/wan_video.py:197
↓ 2 callersMethoddenoising_model
(self)
OmniAvatar/wan_video.py:154
↓ 2 callersMethodencode_prompt
(self, prompt, positive=True)
OmniAvatar/wan_video.py:158
↓ 2 callersMethodencode_video
(self, input_video, tiled=True, tile_size=(34, 34), tile_stride=(18, 16))
OmniAvatar/wan_video.py:192
↓ 2 callersMethodfetch_tokenizer
(self, tokenizer_path=None)
OmniAvatar/prompters/wan_prompter.py:92
↓ 2 callersFunctionflash_attention
(q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, num_heads: int, compatibility_mode=False)
OmniAvatar/models/wan_video_dit.py:32
↓ 2 callersMethodforward
(self, x)
OmniAvatar/models/wan_video_vae.py:67
↓ 2 callersFunctionfp16_clamp
(x)
OmniAvatar/models/wan_video_text_encoder.py:8
↓ 2 callersFunctioninit_weights_on_device
(device = torch.device("meta"), include_buffers :bool = False)
OmniAvatar/utils/io_utils.py:19
↓ 2 callersFunctionlinear_interpolation
Transpose the features to interpolate linearly. Args: features (torch.Tensor): The extracted features to be interpolated. se
OmniAvatar/models/wav2vec.py:196
↓ 2 callersFunctionload_model_from_huggingface_folder
(file_path, model_names, model_classes, torch_dtype, device)
OmniAvatar/models/model_manager.py:44
↓ 2 callersMethodload_models
(self, file_path_list, model_names=None, device=None, torch_dtype=None)
OmniAvatar/models/model_manager.py:364
↓ 2 callersFunctionload_patch_model_from_single_file
(state_dict, model_names, model_classes, extra_kwargs, model_manager, torch_dtype, device)
OmniAvatar/models/model_manager.py:74
↓ 2 callersFunctionmodulate
(x: torch.Tensor, shift: torch.Tensor, scale: torch.Tensor)
OmniAvatar/models/wan_video_dit.py:66
↓ 2 callersFunctionrope_apply
(x, freqs, num_heads)
OmniAvatar/distributed/xdit_context_parallel.py:28
↓ 2 callersFunctionrope_apply
(x, freqs, num_heads)
OmniAvatar/models/wan_video_dit.py:94
↓ 2 callersMethodset_timesteps
(self, num_inference_steps=100, denoising_strength=1.0, training=False, shift=None)
OmniAvatar/schedulers/flow_match.py:18
↓ 2 callersFunctionsplit_state_dict_with_prefix
(state_dict)
OmniAvatar/utils/io_utils.py:187
↓ 2 callersMethodunpatchify
(self, x: torch.Tensor, grid_size: torch.Tensor)
OmniAvatar/models/wan_video_dit.py:325
↓ 2 callersFunctionwhitespace_clean
(text)
OmniAvatar/prompters/wan_prompter.py:17
↓ 2 callersMethodwrite
(self, x)
scripts/inference.py:390
↓ 1 callersMethod__init__
(self, num_inference_steps, rel_l1_thresh, model_id)
OmniAvatar/wan_video.py:294
↓ 1 callersMethod__init__
(self, module: torch.nn.Module, offload_dtype, offload_device, onload_dtype, onload_device, computation_dtype,
OmniAvatar/vram_management/layers.py:12
↓ 1 callersMethod__init__
( self, torch_dtype=torch.float16, device="cuda", model_id_list: List = [],
OmniAvatar/models/model_manager.py:279
↓ 1 callersMethod_clean
(self, text)
OmniAvatar/prompters/wan_prompter.py:74
↓ 1 callersMethod_relative_position_bucket
(self, rel_pos)
OmniAvatar/models/wan_video_text_encoder.py:171
↓ 1 callersMethodadd_lora_to_model
(self, model, lora_rank=4, lora_alpha=4, lora_target_modules="q,k,v,o,ffn.0,ffn.2", init_lora_weights="kaiming
scripts/inference.py:163
↓ 1 callersMethodadd_model_metadata
(self, keys_hash, keys_hash_with_shape, model_names, model_classes, model_resource)
OmniAvatar/models/model_manager.py:118
↓ 1 callersMethodadd_model_metadata
(self, architecture, huggingface_lib, model_name, redirected_architecture)
OmniAvatar/models/model_manager.py:205
↓ 1 callersMethodadd_model_metadata
(self, keys_hash_with_shape, model_name, model_class, extra_kwargs)
OmniAvatar/models/model_manager.py:246
↓ 1 callersFunctioncanonicalize
(text, keep_punctuation_exact_string=None)
OmniAvatar/prompters/wan_prompter.py:23
↓ 1 callersMethodcheck
(self, dit: WanModel, x, t_mod)
OmniAvatar/wan_video.py:314
↓ 1 callersFunctionconvert_namespace_to_dict
将 argparse.Namespace 转为字典，并处理不可序列化对象
OmniAvatar/utils/args_config.py:110
↓ 1 callersMethodenable_cpu_offload
(self)
OmniAvatar/base.py:91
↓ 1 callersMethodenable_vram_management
(self, num_persistent_param_in_dit=None)
OmniAvatar/wan_video.py:38
↓ 1 callersFunctionenable_vram_management_recursively
(model: torch.nn.Module, module_map: dict, module_config: dict, max_num_param=None, overflow_module_config: di
OmniAvatar/vram_management/layers.py:74
↓ 1 callersMethodfetch_models
(self, model_manager: ModelManager)
OmniAvatar/wan_video.py:126
↓ 1 callersMethodfrom_civitai
(self, state_dict)
OmniAvatar/models/wan_video_dit.py:521
↓ 1 callersMethodfrom_diffusers
(self, state_dict)
OmniAvatar/models/wan_video_dit.py:444
↓ 1 callersMethodload_model
(self)
scripts/inference.py:100
↓ 1 callersMethodload_model
(self, file_path, model_names=None, device=None, torch_dtype=None)
OmniAvatar/models/model_manager.py:335
↓ 1 callersFunctionload_single_patch_model_from_single_file
(state_dict, model_name, model_class, base_model, extra_kwargs, torch_dtype, device)
OmniAvatar/models/model_manager.py:62
↓ 1 callersFunctionload_state_dict_from_bin
(file_path, torch_dtype=None)
OmniAvatar/utils/io_utils.py:94
↓ 1 callersFunctionload_state_dict_from_safetensors
(file_path, torch_dtype=None)
OmniAvatar/utils/io_utils.py:84
↓ 1 callersMethodlog_video
( self, lat, prompt, fixed_frame=0, # lat frames image_emb={},
OmniAvatar/wan_video.py:207
↓ 1 callersFunctionmain
()
scripts/inference.py:323
↓ 1 callersFunctionmake_triple
(value: Union[int, Tuple[int, int, int]])
OmniAvatar/models/audio_pack.py:8
↓ 1 callersFunctionmatch_size
(image_size, h, w)
scripts/inference.py:42
↓ 1 callersMethodmerge_latents
(self, value, latents, masks, scales, blur_kernel_size=33, blur_sigma=10.0)
OmniAvatar/base.py:51
↓ 1 callersMethodoffload
(self)
OmniAvatar/vram_management/layers.py:23
↓ 1 callersMethodonload
(self)
OmniAvatar/vram_management/layers.py:28
↓ 1 callersFunctionpad_freqs
(original_tensor, target_len)
OmniAvatar/distributed/xdit_context_parallel.py:16
↓ 1 callersFunctionparse_args
()
OmniAvatar/utils/args_config.py:39
↓ 1 callersFunctionparse_hp_string
(hp_string)
OmniAvatar/utils/args_config.py:7
↓ 1 callersFunctionpatch_tensor_constructor
(fn)
OmniAvatar/utils/io_utils.py:38
↓ 1 callersFunctionprecompute_freqs_cis_3d
(dim: int, end: int = 1024, theta: float = 10000.0)
OmniAvatar/models/wan_video_dit.py:77
↓ 1 callersMethodprepare_extra_input
(self, latents=None)
OmniAvatar/wan_video.py:188
↓ 1 callersMethodprepare_unified_sequence_parallel
(self)
OmniAvatar/wan_video.py:202
↓ 1 callersMethodprocess_prompt
(self, prompt, positive=True)
OmniAvatar/prompters/base_prompter.py:57
↓ 1 callersFunctionread_from_file
(p)
scripts/inference.py:37
↓ 1 callersFunctionreload
重新加载配置文件,不覆盖已有的参数
OmniAvatar/utils/args_config.py:98
↓ 1 callersMethodreparameterize
(self, mu, log_var)
OmniAvatar/models/wan_video_vae.py:577
↓ 1 callersFunctionresize_pad
(image, ori_size, tgt_size)
scripts/inference.py:58
↓ 1 callersFunctionsave_video_as_grid_and_mp4
(video_batch: torch.Tensor, save_path: str, fps: float = 5,prompt=None, prompt_path=None, audio=None, audio_pa
OmniAvatar/utils/io_utils.py:138
↓ 1 callersFunctionsave_wav
(audio, audio_path)
OmniAvatar/utils/io_utils.py:127
↓ 1 callersFunctionset_seed
(seed: int = 42)
scripts/inference.py:30
↓ 1 callersMethodsingle_decode
(self, hidden_state, device)
OmniAvatar/models/wan_video_vae.py:752
↓ 1 callersMethodsingle_encode
(self, video, device)
OmniAvatar/models/wan_video_vae.py:746
↓ 1 callersFunctionsinusoidal_embedding_1d
(dim, position)
OmniAvatar/distributed/xdit_context_parallel.py:10
next →1–100 of 234, ranked by callers