MCPcopy Create free account

hub / github.com/Omni-Avatar/OmniAvatar / types & classes

Types & classes48 in github.com/Omni-Avatar/OmniAvatar

↓ 11 callersClassCausalConv3d
Causal 3d convolusion.
OmniAvatar/models/wan_video_vae.py:33
↓ 6 callersClassResidualBlock
OmniAvatar/models/wan_video_vae.py:198
↓ 5 callersClassRMSNorm
OmniAvatar/models/wan_video_dit.py:102
↓ 5 callersClassRMS_norm
OmniAvatar/models/wan_video_vae.py:55
↓ 4 callersClassAttentionBlock
Causal self-attention with a single head.
OmniAvatar/models/wan_video_vae.py:235
↓ 3 callersClassT5LayerNorm
OmniAvatar/models/wan_video_text_encoder.py:22
↓ 3 callersClassTeaCache
OmniAvatar/wan_video.py:293
↓ 2 callersClassAttentionModule
OmniAvatar/models/wan_video_dit.py:116
↓ 2 callersClassResample
OmniAvatar/models/wan_video_vae.py:82
↓ 2 callersClassT5RelativeEmbedding
OmniAvatar/models/wan_video_text_encoder.py:147
↓ 2 callersClassUpsample
OmniAvatar/models/wan_video_vae.py:73
↓ 1 callersClassAudioPack
OmniAvatar/models/audio_pack.py:14
↓ 1 callersClassCrossAttention
OmniAvatar/models/wan_video_dit.py:152
↓ 1 callersClassDecoder3d
OmniAvatar/models/wan_video_vae.py:379
↓ 1 callersClassDiTBlock
OmniAvatar/models/wan_video_dit.py:198
↓ 1 callersClassEncoder3d
OmniAvatar/models/wan_video_vae.py:276
↓ 1 callersClassFlowMatchScheduler
OmniAvatar/schedulers/flow_match.py:5
↓ 1 callersClassGELU
OmniAvatar/models/wan_video_text_encoder.py:15
↓ 1 callersClassGateModule
OmniAvatar/models/wan_video_dit.py:191
↓ 1 callersClassHead
OmniAvatar/models/wan_video_dit.py:243
↓ 1 callersClassHuggingfaceTokenizer
OmniAvatar/prompters/wan_prompter.py:36
↓ 1 callersClassMLP
OmniAvatar/models/wan_video_dit.py:228
↓ 1 callersClassModelDetectorFromHuggingfaceFolder
OmniAvatar/models/model_manager.py:198
↓ 1 callersClassModelDetectorFromSingleFile
OmniAvatar/models/model_manager.py:110
↓ 1 callersClassModelDetectorFromSplitedSingleFile
OmniAvatar/models/model_manager.py:161
↓ 1 callersClassModelManager
OmniAvatar/models/model_manager.py:278
↓ 1 callersClassNoPrint
scripts/inference.py:389
↓ 1 callersClassSelfAttention
OmniAvatar/models/wan_video_dit.py:126
↓ 1 callersClassT5Attention
OmniAvatar/models/wan_video_text_encoder.py:38
↓ 1 callersClassT5FeedForward
OmniAvatar/models/wan_video_text_encoder.py:92
↓ 1 callersClassT5SelfAttention
OmniAvatar/models/wan_video_text_encoder.py:113
↓ 1 callersClassVideoVAE_
OmniAvatar/models/wan_video_vae.py:492
↓ 1 callersClassWanInferencePipeline
scripts/inference.py:76
↓ 1 callersClassWanModelStateDictConverter
OmniAvatar/models/wan_video_dit.py:440
↓ 1 callersClassWanPrompter
OmniAvatar/prompters/wan_prompter.py:84
↓ 1 callersClassWanTextEncoderStateDictConverter
OmniAvatar/models/wan_video_text_encoder.py:261
↓ 1 callersClassWanVideoPipeline
OmniAvatar/wan_video.py:21
↓ 1 callersClassWanVideoVAEStateDictConverter
OmniAvatar/models/wan_video_vae.py:796
ClassAutoWrappedLinear
OmniAvatar/vram_management/layers.py:41
ClassAutoWrappedModule
OmniAvatar/vram_management/layers.py:11
ClassBasePipeline
OmniAvatar/base.py:8
ClassBasePrompter
OmniAvatar/prompters/base_prompter.py:39
ClassModelDetectorFromPatchedSingleFile
OmniAvatar/models/model_manager.py:239
ClassModelDetectorTemplate
OmniAvatar/models/model_manager.py:98
ClassWanModel
OmniAvatar/models/wan_video_dit.py:259
ClassWanTextEncoder
OmniAvatar/models/wan_video_text_encoder.py:209
ClassWanVideoVAE
OmniAvatar/models/wan_video_vae.py:599
ClassWav2VecModel
Wav2VecModel is a custom model class that extends the Wav2Vec2Model class from the transformers library. It inherits all the functionality o
OmniAvatar/models/wav2vec.py:21