Update README.md
Browse files
README.md
CHANGED
@@ -60,6 +60,10 @@ The core models released in this batch include the following:
|
|
60 |
|
61 |
To load the model with HuggingFace, use the following snippet:
|
62 |
```
|
|
|
|
|
|
|
|
|
63 |
from transformers import AutoModelForSequenceClassification
|
64 |
|
65 |
olmo_model = AutoModelForSequenceClassification.from_pretrained("allenai/OLMo-2-1124-7B-RM")
|
|
|
60 |
|
61 |
To load the model with HuggingFace, use the following snippet:
|
62 |
```
|
63 |
+
# please install from our custom branch
|
64 |
+
# pip install git+https://github.com/vwxyzjn/transformers.git@olmo1124_classification
|
65 |
+
from transformers.models.olmo_1124.modeling_olmo_1124 import Olmo1124ForSequenceClassification, Olmo1124Config
|
66 |
+
AutoModelForSequenceClassification.register(Olmo1124Config, Olmo1124ForSequenceClassification)
|
67 |
from transformers import AutoModelForSequenceClassification
|
68 |
|
69 |
olmo_model = AutoModelForSequenceClassification.from_pretrained("allenai/OLMo-2-1124-7B-RM")
|