MCPcopy Create free account

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

Functions234 in github.com/Omni-Avatar/OmniAvatar

Methodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, infer=False, **kwargs)
OmniAvatar/models/model_manager.py:138
Methodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, **kwargs)
OmniAvatar/models/model_manager.py:178
Methodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, **kwargs)
OmniAvatar/models/model_manager.py:222
Methodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, model_manager=None, **kwargs)
OmniAvatar/models/model_manager.py:261
Methodload_model_from_huggingface_folder
(self, file_path="", model_names=[], model_classes=[])
OmniAvatar/models/model_manager.py:315
Methodload_model_from_single_file
(self, file_path="", state_dict={}, model_names=[], model_classes=[], model_resource=None)
OmniAvatar/models/model_manager.py:303
Methodload_patch_model_from_single_file
(self, file_path="", state_dict={}, model_names=[], model_classes=[], extra_kwargs={})
OmniAvatar/models/model_manager.py:325
Methodload_prompt_extenders
(self,model_manager:ModelManager,extender_classes=[])
OmniAvatar/prompters/base_prompter.py:50
Methodload_prompt_refiners
(self, model_manager: ModelManager, refiner_classes=[])
OmniAvatar/prompters/base_prompter.py:45
Functionload_state_dict_from_folder
(file_path, torch_dtype=None)
OmniAvatar/utils/io_utils.py:67
Methodmatch
(self, file_path="", state_dict={})
OmniAvatar/models/model_manager.py:124
Methodmatch
(self, file_path="", state_dict={})
OmniAvatar/models/model_manager.py:166
Methodmatch
(self, file_path="", state_dict={})
OmniAvatar/models/model_manager.py:209
Methodmatch
(self, file_path="", state_dict={})
OmniAvatar/models/model_manager.py:250
Methodoffload
(self)
OmniAvatar/vram_management/layers.py:55
Methodonload
(self)
OmniAvatar/vram_management/layers.py:60
Methodpreprocess_images
(self, images)
OmniAvatar/base.py:35
Functionregister_empty_buffer
(module, name, buffer, persistent=True)
OmniAvatar/utils/io_utils.py:33
Functionregister_empty_parameter
(module, name, param)
OmniAvatar/utils/io_utils.py:25
Methodreturn_to_timestep
(self, timestep, sample, sample_stablized)
OmniAvatar/schedulers/flow_match.py:53
Methodsample
(self, imgs, deterministic=False)
OmniAvatar/models/wan_video_vae.py:582
Functionsearch_for_files
(folder, extensions)
OmniAvatar/utils/io_utils.py:220
Functionshard_model
( model, device_id, param_dtype=torch.bfloat16, reduce_dtype=torch.float32, buffer_dtype=t
OmniAvatar/distributed/fsdp.py:12
Methodstate_dict_converter
()
OmniAvatar/models/wan_video_text_encoder.py:257
Methodstate_dict_converter
()
OmniAvatar/models/wan_video_vae.py:792
Methodtensor2video
(self, frames)
OmniAvatar/wan_video.py:181
Functiontokenize_long_prompt
(tokenizer, prompt, max_length=None)
OmniAvatar/prompters/base_prompter.py:6
Methodtraining_target
(self, sample, noise, timestep)
OmniAvatar/schedulers/flow_match.py:71
Methodtraining_weight
(self, timestep)
OmniAvatar/schedulers/flow_match.py:76
Functionusp_attn_forward
(self, x, freqs)
OmniAvatar/distributed/xdit_context_parallel.py:115
Functionusp_dit_forward
(self, x: torch.Tensor, timestep: torch.Tensor, context: torch.Tensor,
OmniAvatar/distributed/xdit_context_parallel.py:46
Methodvae_output_to_image
(self, vae_output)
OmniAvatar/base.py:39
Methodvae_output_to_video
(self, vae_output)
OmniAvatar/base.py:45
Functionwrapper
(*args, **kwargs)
OmniAvatar/utils/io_utils.py:39
← previous201–234 of 234, ranked by callers