ailuntz commited on
Commit
68dfbde
·
verified ·
1 Parent(s): ad93f51

Add default bf16 config.json

Browse files
Files changed (1) hide show
  1. config.json +129 -0
config.json ADDED
@@ -0,0 +1,129 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "format": "songgeneration-mlx-audiolm",
3
+ "source": "tencent/SongGeneration",
4
+ "official_code": "https://github.com/tencent-ailab/songgeneration",
5
+ "variant": "v2-medium",
6
+ "precision": "bfloat16",
7
+ "runtime": {
8
+ "hidden_size": 1536,
9
+ "intermediate_size": 8960,
10
+ "num_heads": 12,
11
+ "num_layers": 28,
12
+ "num_layers_sub": 12,
13
+ "code_depth": 3,
14
+ "code_size": 16384,
15
+ "prompt_len": 10,
16
+ "frame_rate": 25,
17
+ "max_position_embeddings": 10000,
18
+ "max_position_embeddings_sub": 10000,
19
+ "rope_theta": 500000.0,
20
+ "rope_theta_sub": 500000.0,
21
+ "cfg_coef": 1.5
22
+ },
23
+ "components": {
24
+ "audiolm": "converted to MLX safetensors",
25
+ "qwen_tokenizer": "copied for lyric/style tokenization",
26
+ "audio_decoder": "use official PyTorch Flow1dVAE/separate tokenizer bridge for now"
27
+ },
28
+ "skipped_keys": [
29
+ "audiolm.out_norm.weight",
30
+ "audiolm.out_norm.bias",
31
+ "audiolm.transformer2.lm_head.weight"
32
+ ],
33
+ "weight_files": [
34
+ "model-00001-of-00094.safetensors",
35
+ "model-00002-of-00094.safetensors",
36
+ "model-00003-of-00094.safetensors",
37
+ "model-00004-of-00094.safetensors",
38
+ "model-00005-of-00094.safetensors",
39
+ "model-00006-of-00094.safetensors",
40
+ "model-00007-of-00094.safetensors",
41
+ "model-00008-of-00094.safetensors",
42
+ "model-00009-of-00094.safetensors",
43
+ "model-00010-of-00094.safetensors",
44
+ "model-00011-of-00094.safetensors",
45
+ "model-00012-of-00094.safetensors",
46
+ "model-00013-of-00094.safetensors",
47
+ "model-00014-of-00094.safetensors",
48
+ "model-00015-of-00094.safetensors",
49
+ "model-00016-of-00094.safetensors",
50
+ "model-00017-of-00094.safetensors",
51
+ "model-00018-of-00094.safetensors",
52
+ "model-00019-of-00094.safetensors",
53
+ "model-00020-of-00094.safetensors",
54
+ "model-00021-of-00094.safetensors",
55
+ "model-00022-of-00094.safetensors",
56
+ "model-00023-of-00094.safetensors",
57
+ "model-00024-of-00094.safetensors",
58
+ "model-00025-of-00094.safetensors",
59
+ "model-00026-of-00094.safetensors",
60
+ "model-00027-of-00094.safetensors",
61
+ "model-00028-of-00094.safetensors",
62
+ "model-00029-of-00094.safetensors",
63
+ "model-00030-of-00094.safetensors",
64
+ "model-00031-of-00094.safetensors",
65
+ "model-00032-of-00094.safetensors",
66
+ "model-00033-of-00094.safetensors",
67
+ "model-00034-of-00094.safetensors",
68
+ "model-00035-of-00094.safetensors",
69
+ "model-00036-of-00094.safetensors",
70
+ "model-00037-of-00094.safetensors",
71
+ "model-00038-of-00094.safetensors",
72
+ "model-00039-of-00094.safetensors",
73
+ "model-00040-of-00094.safetensors",
74
+ "model-00041-of-00094.safetensors",
75
+ "model-00042-of-00094.safetensors",
76
+ "model-00043-of-00094.safetensors",
77
+ "model-00044-of-00094.safetensors",
78
+ "model-00045-of-00094.safetensors",
79
+ "model-00046-of-00094.safetensors",
80
+ "model-00047-of-00094.safetensors",
81
+ "model-00048-of-00094.safetensors",
82
+ "model-00049-of-00094.safetensors",
83
+ "model-00050-of-00094.safetensors",
84
+ "model-00051-of-00094.safetensors",
85
+ "model-00052-of-00094.safetensors",
86
+ "model-00053-of-00094.safetensors",
87
+ "model-00054-of-00094.safetensors",
88
+ "model-00055-of-00094.safetensors",
89
+ "model-00056-of-00094.safetensors",
90
+ "model-00057-of-00094.safetensors",
91
+ "model-00058-of-00094.safetensors",
92
+ "model-00059-of-00094.safetensors",
93
+ "model-00060-of-00094.safetensors",
94
+ "model-00061-of-00094.safetensors",
95
+ "model-00062-of-00094.safetensors",
96
+ "model-00063-of-00094.safetensors",
97
+ "model-00064-of-00094.safetensors",
98
+ "model-00065-of-00094.safetensors",
99
+ "model-00066-of-00094.safetensors",
100
+ "model-00067-of-00094.safetensors",
101
+ "model-00068-of-00094.safetensors",
102
+ "model-00069-of-00094.safetensors",
103
+ "model-00070-of-00094.safetensors",
104
+ "model-00071-of-00094.safetensors",
105
+ "model-00072-of-00094.safetensors",
106
+ "model-00073-of-00094.safetensors",
107
+ "model-00074-of-00094.safetensors",
108
+ "model-00075-of-00094.safetensors",
109
+ "model-00076-of-00094.safetensors",
110
+ "model-00077-of-00094.safetensors",
111
+ "model-00078-of-00094.safetensors",
112
+ "model-00079-of-00094.safetensors",
113
+ "model-00080-of-00094.safetensors",
114
+ "model-00081-of-00094.safetensors",
115
+ "model-00082-of-00094.safetensors",
116
+ "model-00083-of-00094.safetensors",
117
+ "model-00084-of-00094.safetensors",
118
+ "model-00085-of-00094.safetensors",
119
+ "model-00086-of-00094.safetensors",
120
+ "model-00087-of-00094.safetensors",
121
+ "model-00088-of-00094.safetensors",
122
+ "model-00089-of-00094.safetensors",
123
+ "model-00090-of-00094.safetensors",
124
+ "model-00091-of-00094.safetensors",
125
+ "model-00092-of-00094.safetensors",
126
+ "model-00093-of-00094.safetensors",
127
+ "model-00094-of-00094.safetensors"
128
+ ]
129
+ }