MCPcopy Create free account
hub / github.com/FunAudioLLM/ThinkSound / __init__

Method __init__

ThinkSound/models/mmdit.py:29–198  ·  view source on GitHub ↗
(self,
                 *,
                 latent_dim: int,
                 clip_dim: int,
                 sync_dim: int,
                 text_dim: int,
                 hidden_dim: int,
                 depth: int,
                 fused_depth: int,
                 num_heads: int,
                 mlp_ratio: float = 4.0,
                 latent_seq_len: int,
                 clip_seq_len: int,
                 sync_seq_len: int,
                 text_seq_len: int = 77,
                 latent_mean: Optional[torch.Tensor] = None,
                 latent_std: Optional[torch.Tensor] = None,
                 empty_string_feat: Optional[torch.Tensor] = None,
                 v2: bool = False,
                 kernel_size: int = 7,
                 sync_kernel: int = 7,
                 use_inpaint: bool = False,
                 use_mlp: bool = False,
                 cross_attend: bool = False,
                 add_video: bool = False,
                 triple_fusion: bool = False,
                 gated_video: bool = False)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 9

initialize_weightsMethod · 0.95
initialize_rotationsMethod · 0.95
ChannelLastConv1dClass · 0.85
ConvMLPClass · 0.85
MLPClass · 0.85
FinalBlockClass · 0.85
TimestepEmbedderClass · 0.85
JointBlockClass · 0.85
MMDitSingleBlockClass · 0.85

Tested by

no test coverage detected