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

Method __init__

ThinkSound/models/dit.py:14–155  ·  view source on GitHub ↗
(self, 
        io_channels=32, 
        patch_size=1,
        embed_dim=768,
        cond_token_dim=0,
        project_cond_tokens=True,
        global_cond_dim=0,
        project_global_cond=True,
        input_concat_dim=0,
        prepend_cond_dim=0,
        cond_ctx_dim=0,
        depth=12,
        num_heads=8,
        transformer_type: tp.Literal["x-transformers", "continuous_transformer","mm_transformer"] = "x-transformers",
        global_cond_type: tp.Literal["prepend", "adaLN"] = "prepend",
        frac_lengths_mask = (0.7, 1.),
        ctx_drop: float = 0.1,
        add_token_dim=0,
        use_mlp=False,
        **kwargs)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

FourierFeaturesClass · 0.85
ConvMLPClass · 0.85

Tested by

no test coverage detected