Update configuration_mistral.py
Browse files- configuration_mistral.py +1 -1
configuration_mistral.py
CHANGED
@@ -142,7 +142,7 @@ class MistralConfig(PretrainedConfig):
|
|
142 |
use_shallow_talk=False,
|
143 |
use_complex_think_head=False,
|
144 |
use_complex_talk_head=True,
|
145 |
-
use_weighted_talk_head=
|
146 |
**kwargs,
|
147 |
):
|
148 |
self.vocab_size = vocab_size
|
|
|
142 |
use_shallow_talk=False,
|
143 |
use_complex_think_head=False,
|
144 |
use_complex_talk_head=True,
|
145 |
+
use_weighted_talk_head=False,
|
146 |
**kwargs,
|
147 |
):
|
148 |
self.vocab_size = vocab_size
|