MCPcopy Create free account

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

Functions234 in github.com/Omni-Avatar/OmniAvatar

⨍ Functions 234◇ Types & classes 48
↓ 1 callersFunctionsinusoidal_embedding_1d
(dim, position)
OmniAvatar/models/wan_video_dit.py:70
↓ 1 callersFunctionsmart_load_weights
(model, ckpt_state_dict)
OmniAvatar/utils/io_utils.py:102
↓ 1 callersMethodstate_dict_converter
()
OmniAvatar/models/wan_video_dit.py:436
↓ 1 callersMethodstep
(self, model_output, timestep, sample, to_final=False, **kwargs)
OmniAvatar/schedulers/flow_match.py:40
↓ 1 callersMethodstore
(self, hidden_states)
OmniAvatar/wan_video.py:336
↓ 1 callersMethodtiled_decode
(self, hidden_states, device, tile_size, tile_stride)
OmniAvatar/models/wan_video_vae.py:643
↓ 1 callersMethodtiled_encode
(self, video, device, tile_size, tile_stride)
OmniAvatar/models/wan_video_vae.py:695
Method__call__
(self, sequence, **kwargs)
OmniAvatar/prompters/wan_prompter.py:48
Method__init__
(self, args)
scripts/inference.py:77
Method__init__
(self, device="cuda", torch_dtype=torch.float16, height_division_factor=64, width_division_factor=64)
OmniAvatar/base.py:10
Method__init__
(self, device="cuda", torch_dtype=torch.float16, tokenizer_path=None)
OmniAvatar/wan_video.py:23
Method__init__
(self, module: torch.nn.Linear, offload_dtype, offload_device, onload_dtype, onload_device, computation_dtype,
OmniAvatar/vram_management/layers.py:42
Method__init__
(self, name, seq_len=None, clean=None, **kwargs)
OmniAvatar/prompters/wan_prompter.py:38
Method__init__
(self, tokenizer_path=None, text_len=512)
OmniAvatar/prompters/wan_prompter.py:86
Method__init__
(self)
OmniAvatar/prompters/base_prompter.py:40
Method__init__
(self, num_inference_steps=100, num_train_timesteps=1000, shift=3.0, sigma_max=1.0, sigma_min=0.003/1.002, inv
OmniAvatar/schedulers/flow_match.py:7
Method__init__
(self, dim, dim_attn, num_heads, dropout=0.1)
OmniAvatar/models/wan_video_text_encoder.py:40
Method__init__
(self, dim, dim_ffn, dropout=0.1)
OmniAvatar/models/wan_video_text_encoder.py:94
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
OmniAvatar/models/wan_video_text_encoder.py:115
Method__init__
(self, num_buckets, num_heads, bidirectional, max_dist=128)
OmniAvatar/models/wan_video_text_encoder.py:149
Method__init__
(self, vocab=256384, dim=4096, dim_attn=4096,
OmniAvatar/models/wan_video_text_encoder.py:211
Method__init__
(self)
OmniAvatar/models/wan_video_text_encoder.py:262
Method__init__
(self)
OmniAvatar/models/model_manager.py:99
Method__init__
(self, model_loader_configs=[])
OmniAvatar/models/model_manager.py:111
Method__init__
(self, model_loader_configs=[])
OmniAvatar/models/model_manager.py:162
Method__init__
(self, model_loader_configs=[])
OmniAvatar/models/model_manager.py:199
Method__init__
(self, model_loader_configs=[])
OmniAvatar/models/model_manager.py:240
Method__init__
(self, *args, **kwargs)
OmniAvatar/models/wan_video_vae.py:38
Method__init__
(self, dim, mode)
OmniAvatar/models/wan_video_vae.py:84
Method__init__
(self, in_dim, out_dim, dropout=0.0)
OmniAvatar/models/wan_video_vae.py:200
Method__init__
(self, dim)
OmniAvatar/models/wan_video_vae.py:240
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
OmniAvatar/models/wan_video_vae.py:278
Method__init__
(self, dim=128, z_dim=4, dim_mult=[1, 2, 4, 4],
OmniAvatar/models/wan_video_vae.py:381
Method__init__
(self, dim=96, z_dim=16, dim_mult=[1, 2, 4, 4],
OmniAvatar/models/wan_video_vae.py:494
Method__init__
(self, z_dim=16)
OmniAvatar/models/wan_video_vae.py:601
Method__init__
(self)
OmniAvatar/models/wan_video_vae.py:798
Method__init__
( self, in_channels: int, patch_size: Union[int, Tuple[int, int, int]],
OmniAvatar/models/audio_pack.py:15
Method__init__
(self, dim, eps=1e-5)
OmniAvatar/models/wan_video_dit.py:103
Method__init__
(self, num_heads)
OmniAvatar/models/wan_video_dit.py:117
Method__init__
(self, dim: int, num_heads: int, eps: float = 1e-6)
OmniAvatar/models/wan_video_dit.py:127
Method__init__
(self, dim: int, num_heads: int, eps: float = 1e-6, has_image_input: bool = False)
OmniAvatar/models/wan_video_dit.py:153
Method__init__
(self,)
OmniAvatar/models/wan_video_dit.py:192
Method__init__
(self, has_image_input: bool, dim: int, num_heads: int, ffn_dim: int, eps: float = 1e-6)
OmniAvatar/models/wan_video_dit.py:199
Method__init__
(self, dim: int, out_dim: int, patch_size: Tuple[int, int, int], eps: float)
OmniAvatar/models/wan_video_dit.py:244
Method__init__
( self, dim: int, in_dim: int, ffn_dim: int, out_dim: int, tex
OmniAvatar/models/wan_video_dit.py:260
Method__init__
(self)
OmniAvatar/models/wan_video_dit.py:441
Methodadd_noise
(self, original_samples, noise, timestep)
OmniAvatar/schedulers/flow_match.py:62
Functionblock_causal_mask
(x, block_size)
OmniAvatar/models/wan_video_vae.py:19
Methodcheck_resize_height_width
(self, height, width)
OmniAvatar/base.py:20
Methodcontrol_noise_via_local_prompts
(self, prompt_emb_global, prompt_emb_locals, masks, mask_scales, inference_callback, special_kwargs=None, spec
OmniAvatar/base.py:66
Methodcreate_custom_forward
(module)
OmniAvatar/models/wan_video_dit.py:367
Functioncustom_forward
(*inputs)
OmniAvatar/distributed/xdit_context_parallel.py:75
Methodcustom_forward
(*inputs)
OmniAvatar/models/wan_video_dit.py:368
Methoddecode
(self, hidden_states, device, tiled=False, tile_size=(34, 34), tile_stride=(18, 16))
OmniAvatar/models/wan_video_vae.py:776
Methodencode
(self, videos, device, tiled=False, tile_size=(34, 34), tile_stride=(18, 16))
OmniAvatar/models/wan_video_vae.py:758
Methodencode_image
(self, image, num_frames, height, width)
OmniAvatar/wan_video.py:163
Methodencode_prompt
(self, prompt, positive=True, device="cuda")
OmniAvatar/prompters/wan_prompter.py:99
Methodextend_prompt
(self, prompt, local_prompts, masks, mask_scales)
OmniAvatar/base.py:79
Methodextend_prompt
(self, prompt:str, positive=True)
OmniAvatar/prompters/base_prompter.py:66
Methodfeature_extract
Extracts features from the input values and returns the extracted features. Parameters: input_values (torch.Tensor): The inp
OmniAvatar/models/wav2vec.py:112
Methodfetch_models
(self, text_encoder: WanTextEncoder = None)
OmniAvatar/prompters/wan_prompter.py:96
Methodflush
(self)
scripts/inference.py:392
Methodforward
(self, prompt, image_path=None, audio_path=None, seq_len=10
scripts/inference.py:189
Methodforward
(self, *args, **kwargs)
OmniAvatar/vram_management/layers.py:33
Methodforward
(self, x, *args, **kwargs)
OmniAvatar/vram_management/layers.py:65
Methodforward
(self, x)
OmniAvatar/models/wan_video_text_encoder.py:17
Methodforward
(self, x)
OmniAvatar/models/wan_video_text_encoder.py:30
Methodforward
x: [B, L1, C]. context: [B, L2, C] or None. mask: [B, L2] or [B, L1, L2] or None.
OmniAvatar/models/wan_video_text_encoder.py:55
Methodforward
(self, x)
OmniAvatar/models/wan_video_text_encoder.py:105
Methodforward
(self, x, mask=None, pos_bias=None)
OmniAvatar/models/wan_video_text_encoder.py:139
Methodforward
(self, lq, lk)
OmniAvatar/models/wan_video_text_encoder.py:159
Methodforward
(self, ids, mask=None)
OmniAvatar/models/wan_video_text_encoder.py:245
Methodforward
(self, x, cache_x=None)
OmniAvatar/models/wan_video_vae.py:44
Methodforward
Fix bfloat16 support for nearest neighbor interpolation.
OmniAvatar/models/wan_video_vae.py:75
Methodforward
(self, x, feat_cache=None, feat_idx=[0])
OmniAvatar/models/wan_video_vae.py:120
Methodforward
(self, x, feat_cache=None, feat_idx=[0])
OmniAvatar/models/wan_video_vae.py:214
Methodforward
(self, x)
OmniAvatar/models/wan_video_vae.py:252
Methodforward
(self, x, feat_cache=None, feat_idx=[0])
OmniAvatar/models/wan_video_vae.py:328
Methodforward
(self, x, feat_cache=None, feat_idx=[0])
OmniAvatar/models/wan_video_vae.py:432
Methodforward
(self, x)
OmniAvatar/models/wan_video_vae.py:519
Methodforward
( self, vid: torch.Tensor, )
OmniAvatar/models/audio_pack.py:31
Methodforward
Forward pass of the Wav2Vec model. Args: self: The instance of the model. input_values: The input values (wa
OmniAvatar/models/wav2vec.py:42
Methodforward
(self, x)
OmniAvatar/models/wan_video_dit.py:111
Methodforward
(self, q, k, v)
OmniAvatar/models/wan_video_dit.py:121
Methodforward
(self, x, freqs)
OmniAvatar/models/wan_video_dit.py:142
Methodforward
(self, x: torch.Tensor, y: torch.Tensor)
OmniAvatar/models/wan_video_dit.py:173
Methodforward
(self, x, gate, residual)
OmniAvatar/models/wan_video_dit.py:195
Methodforward
(self, x, context, t_mod, freqs)
OmniAvatar/models/wan_video_dit.py:216
Methodforward
(self, x)
OmniAvatar/models/wan_video_dit.py:239
Methodforward
(self, x, t_mod)
OmniAvatar/models/wan_video_dit.py:252
Methodforward
(self, x: torch.Tensor, timestep: torch.Tensor, context: torch
OmniAvatar/models/wan_video_dit.py:332
Functionfree_model
(model)
OmniAvatar/distributed/fsdp.py:37
Methodfrom_civitai
(self, state_dict)
OmniAvatar/models/wan_video_text_encoder.py:268
Methodfrom_civitai
(self, state_dict)
OmniAvatar/models/wan_video_vae.py:801
Methodfrom_diffusers
(self, state_dict)
OmniAvatar/models/wan_video_text_encoder.py:265
Methodgenerate_noise
(self, shape, seed=None, device="cpu", dtype=torch.float16)
OmniAvatar/base.py:124
Methodinit_weight
(self, conv)
OmniAvatar/models/wan_video_vae.py:176
Methodinit_weight2
(self, conv)
OmniAvatar/models/wan_video_vae.py:187
Functioninit_weights
(m)
OmniAvatar/models/wan_video_text_encoder.py:192
Functionis_zero_stage_3
(trainer)
OmniAvatar/utils/io_utils.py:174
← previousnext →101–200 of 234, ranked by callers