Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Omni-Avatar/OmniAvatar
/ types & classes
Types & classes
48 in github.com/Omni-Avatar/OmniAvatar
⨍
Functions
234
◇
Types & classes
48
↓ 11 callers
Class
CausalConv3d
Causal 3d convolusion.
OmniAvatar/models/wan_video_vae.py:33
↓ 6 callers
Class
ResidualBlock
OmniAvatar/models/wan_video_vae.py:198
↓ 5 callers
Class
RMSNorm
OmniAvatar/models/wan_video_dit.py:102
↓ 5 callers
Class
RMS_norm
OmniAvatar/models/wan_video_vae.py:55
↓ 4 callers
Class
AttentionBlock
Causal self-attention with a single head.
OmniAvatar/models/wan_video_vae.py:235
↓ 3 callers
Class
T5LayerNorm
OmniAvatar/models/wan_video_text_encoder.py:22
↓ 3 callers
Class
TeaCache
OmniAvatar/wan_video.py:293
↓ 2 callers
Class
AttentionModule
OmniAvatar/models/wan_video_dit.py:116
↓ 2 callers
Class
Resample
OmniAvatar/models/wan_video_vae.py:82
↓ 2 callers
Class
T5RelativeEmbedding
OmniAvatar/models/wan_video_text_encoder.py:147
↓ 2 callers
Class
Upsample
OmniAvatar/models/wan_video_vae.py:73
↓ 1 callers
Class
AudioPack
OmniAvatar/models/audio_pack.py:14
↓ 1 callers
Class
CrossAttention
OmniAvatar/models/wan_video_dit.py:152
↓ 1 callers
Class
Decoder3d
OmniAvatar/models/wan_video_vae.py:379
↓ 1 callers
Class
DiTBlock
OmniAvatar/models/wan_video_dit.py:198
↓ 1 callers
Class
Encoder3d
OmniAvatar/models/wan_video_vae.py:276
↓ 1 callers
Class
FlowMatchScheduler
OmniAvatar/schedulers/flow_match.py:5
↓ 1 callers
Class
GELU
OmniAvatar/models/wan_video_text_encoder.py:15
↓ 1 callers
Class
GateModule
OmniAvatar/models/wan_video_dit.py:191
↓ 1 callers
Class
Head
OmniAvatar/models/wan_video_dit.py:243
↓ 1 callers
Class
HuggingfaceTokenizer
OmniAvatar/prompters/wan_prompter.py:36
↓ 1 callers
Class
MLP
OmniAvatar/models/wan_video_dit.py:228
↓ 1 callers
Class
ModelDetectorFromHuggingfaceFolder
OmniAvatar/models/model_manager.py:198
↓ 1 callers
Class
ModelDetectorFromSingleFile
OmniAvatar/models/model_manager.py:110
↓ 1 callers
Class
ModelDetectorFromSplitedSingleFile
OmniAvatar/models/model_manager.py:161
↓ 1 callers
Class
ModelManager
OmniAvatar/models/model_manager.py:278
↓ 1 callers
Class
NoPrint
scripts/inference.py:389
↓ 1 callers
Class
SelfAttention
OmniAvatar/models/wan_video_dit.py:126
↓ 1 callers
Class
T5Attention
OmniAvatar/models/wan_video_text_encoder.py:38
↓ 1 callers
Class
T5FeedForward
OmniAvatar/models/wan_video_text_encoder.py:92
↓ 1 callers
Class
T5SelfAttention
OmniAvatar/models/wan_video_text_encoder.py:113
↓ 1 callers
Class
VideoVAE_
OmniAvatar/models/wan_video_vae.py:492
↓ 1 callers
Class
WanInferencePipeline
scripts/inference.py:76
↓ 1 callers
Class
WanModelStateDictConverter
OmniAvatar/models/wan_video_dit.py:440
↓ 1 callers
Class
WanPrompter
OmniAvatar/prompters/wan_prompter.py:84
↓ 1 callers
Class
WanTextEncoderStateDictConverter
OmniAvatar/models/wan_video_text_encoder.py:261
↓ 1 callers
Class
WanVideoPipeline
OmniAvatar/wan_video.py:21
↓ 1 callers
Class
WanVideoVAEStateDictConverter
OmniAvatar/models/wan_video_vae.py:796
Class
AutoWrappedLinear
OmniAvatar/vram_management/layers.py:41
Class
AutoWrappedModule
OmniAvatar/vram_management/layers.py:11
Class
BasePipeline
OmniAvatar/base.py:8
Class
BasePrompter
OmniAvatar/prompters/base_prompter.py:39
Class
ModelDetectorFromPatchedSingleFile
OmniAvatar/models/model_manager.py:239
Class
ModelDetectorTemplate
OmniAvatar/models/model_manager.py:98
Class
WanModel
OmniAvatar/models/wan_video_dit.py:259
Class
WanTextEncoder
OmniAvatar/models/wan_video_text_encoder.py:209
Class
WanVideoVAE
OmniAvatar/models/wan_video_vae.py:599
Class
Wav2VecModel
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