Use correct class name for `OlmoForCausalLM`
Browse filesThis error was introduced in https://huggingface.co/allenai/OLMo-7B/commit/e3131094a01dfd83915c06d1bd3411b1f2327b86
- config.json +1 -1
config.json
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
"alibi": false,
|
| 4 |
"alibi_bias_max": 8.0,
|
| 5 |
"architectures": [
|
| 6 |
-
"
|
| 7 |
],
|
| 8 |
"attention_dropout": 0.0,
|
| 9 |
"attention_layer_norm": false,
|
|
|
|
| 3 |
"alibi": false,
|
| 4 |
"alibi_bias_max": 8.0,
|
| 5 |
"architectures": [
|
| 6 |
+
"OlmoForCausalLM"
|
| 7 |
],
|
| 8 |
"attention_dropout": 0.0,
|
| 9 |
"attention_layer_norm": false,
|