DestaModel
custom_code
File size: 2,571 Bytes
8ab1e14
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
{
  "llama_config": {
    "_name_or_path": "meta-llama/Meta-Llama-3-8B-Instruct",
    "architectures": [
      "LlamaForCausalLM"
    ],
    "bos_token_id": 128000,
    "eos_token_id": 128009,
    "intermediate_size": 14336,
    "max_position_embeddings": 8192,
    "model_type": "llama",
    "num_key_value_heads": 8,
    "rms_norm_eps": 1e-05,
    "rope_theta": 500000.0,
    "torch_dtype": "bfloat16",
    "vocab_size": 128256
  },
  "auto_map": {
    "AutoConfig": "modeling_desta.Desta2Config",
    "AutoModel": "modeling_desta.DestaModel"
  },
  "llama_model_id": "meta-llama/Meta-Llama-3-8B-Instruct",
  "model_type": "DestaModel",
  "prompt_size": 64,
  "transformers_version": "4.41.1",
  "whisper_config": {
    "_name_or_path": "openai/whisper-small",
    "architectures": [
      "WhisperForConditionalGeneration"
    ],
    "begin_suppress_tokens": [
      220,
      50257
    ],
    "bos_token_id": 50257,
    "d_model": 768,
    "decoder_attention_heads": 12,
    "decoder_ffn_dim": 3072,
    "decoder_layers": 12,
    "decoder_start_token_id": 50258,
    "encoder_attention_heads": 12,
    "encoder_ffn_dim": 3072,
    "encoder_layers": 12,
    "eos_token_id": 50257,
    "forced_decoder_ids": [
      [
        1,
        50259
      ],
      [
        2,
        50359
      ],
      [
        3,
        50363
      ]
    ],
    "max_length": 448,
    "model_type": "whisper",
    "num_hidden_layers": 12,
    "pad_token_id": 50257,
    "suppress_tokens": [
      1,
      2,
      7,
      8,
      9,
      10,
      14,
      25,
      26,
      27,
      28,
      29,
      31,
      58,
      59,
      60,
      61,
      62,
      63,
      90,
      91,
      92,
      93,
      359,
      503,
      522,
      542,
      873,
      893,
      902,
      918,
      922,
      931,
      1350,
      1853,
      1982,
      2460,
      2627,
      3246,
      3253,
      3268,
      3536,
      3846,
      3961,
      4183,
      4667,
      6585,
      6647,
      7273,
      9061,
      9383,
      10428,
      10929,
      11938,
      12033,
      12331,
      12562,
      13793,
      14157,
      14635,
      15265,
      15618,
      16553,
      16604,
      18362,
      18956,
      20075,
      21675,
      22520,
      26130,
      26161,
      26435,
      28279,
      29464,
      31650,
      32302,
      32470,
      36865,
      42863,
      47425,
      49870,
      50254,
      50258,
      50360,
      50361,
      50362
    ],
    "torch_dtype": "float32"
  },
  "whisper_model_id": "openai/whisper-small"
}