vwxyzjn commited on
Commit
216f4ad
·
verified ·
1 Parent(s): 5f49edf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
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")