File size: 1,845 Bytes
eb41300
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
{
  "model_type": "diffusion_cond",
  "sample_rate": 44100,
  "model": {
    "conditioning": {
      "cond_dim": 768,
      "configs": [
        {
          "id": "prompt",
          "type": "qwen_text",
          "config": {
            "model_path": "text_encoder",
            "hidden_dim": 1024,
            "max_length": 512,
            "enable_grad": false,
            "project_out": true,
            "padding_mode": "zero",
            "caption_region_embedding": false,
            "source_region_embedding": false,
            "sceneplan_role_embedding": true,
            "sceneplan_max_sources": 4,
            "sceneplan_role_scale": 1.0,
            "fail_on_truncation": true
          }
        },
        {
          "id": "sceneplan_44",
          "type": "sceneplan_44",
          "config": {
            "output_dim": 256,
            "max_sources": 4,
            "trajectory_feature_dim": 5,
            "event_embedding_dim": 32,
            "trajectory_embedding_dim": 32
          }
        }
      ],
      "pre_encoded_keys": [
        "source_foa_latent"
      ]
    },
    "diffusion": {
      "diffusion_objective": "rectified_flow",
      "mask_padding_attention": true,
      "cross_attention_cond_ids": [
        "prompt"
      ],
      "global_cond_ids": [],
      "type": "dit",
      "config": {
        "io_channels": 64,
        "embed_dim": 1024,
        "depth": 15,
        "num_heads": 16,
        "cond_token_dim": 768,
        "global_cond_dim": 0,
        "project_cond_tokens": true,
        "transformer_type": "continuous_transformer",
        "input_concat_dim": 320,
        "input_concat_cond_cfg": true,
        "require_explicit_negative_input_concat": true
      },
      "input_concat_ids": [
        "sceneplan_44",
        "source_foa_latent"
      ]
    },
    "io_channels": 64
  }
}