Keep the readout in float32
Browse files- config.json +3 -0
config.json
CHANGED
|
@@ -206,6 +206,9 @@
|
|
| 206 |
"dtype": "bfloat16",
|
| 207 |
"hidden_size": 5120,
|
| 208 |
"initializer_range": 0.02,
|
|
|
|
|
|
|
|
|
|
| 209 |
"model_type": "jev",
|
| 210 |
"num_verbalizers": 255,
|
| 211 |
"offsets": {
|
|
|
|
| 206 |
"dtype": "bfloat16",
|
| 207 |
"hidden_size": 5120,
|
| 208 |
"initializer_range": 0.02,
|
| 209 |
+
"keep_in_fp32_modules": [
|
| 210 |
+
"^proj"
|
| 211 |
+
],
|
| 212 |
"model_type": "jev",
|
| 213 |
"num_verbalizers": 255,
|
| 214 |
"offsets": {
|