add head_dim
Browse filesFix bug of vllm version 0.8.5:
```
self.q_size = self.num_heads * self.head_dim
TypeError: unsupported operand type(s) for *: 'int' and 'NoneType'
```
- config.json +1 -1
config.json
CHANGED
@@ -14,7 +14,7 @@
|
|
14 |
"MistralForCausalLM"
|
15 |
],
|
16 |
"attention_dropout": 0.0,
|
17 |
-
"head_dim":
|
18 |
"hidden_act": "silu",
|
19 |
"hidden_size": 4096,
|
20 |
"initializer_range": 0.02,
|
|
|
14 |
"MistralForCausalLM"
|
15 |
],
|
16 |
"attention_dropout": 0.0,
|
17 |
+
"head_dim": 128,
|
18 |
"hidden_act": "silu",
|
19 |
"hidden_size": 4096,
|
20 |
"initializer_range": 0.02,
|