AutoModelForSequenceClassification had wrong capitalization
Browse filesAutoModelForSequenceClassification needed to have "For" capitalized, and was spelled as AutoModelforSequenceClassification
README.md
CHANGED
@@ -36,9 +36,9 @@ Note that this model is primarily aimed at being fine-tuned on tasks that use th
|
|
36 |
You can fine-tune it for SequenceClassification, SequentialSentenceClassification, and MultipleChoice down-stream tasks:
|
37 |
|
38 |
```python
|
39 |
-
from transformers import AutoTokenizer,
|
40 |
tokenizer = AutoTokenizer.from_pretrained("kiddothe2b/adhoc-hierarchical-transformer-base-4096", trust_remote_code=True)
|
41 |
-
doc_classifier =
|
42 |
```
|
43 |
|
44 |
Note: If you wish to use a fully pre-trained HAT model, you have to use [kiddothe2b/hierarchical-transformer-base-4096](https://huggingface.co/kiddothe2b/hierarchical-transformer-base-4096).
|
|
|
36 |
You can fine-tune it for SequenceClassification, SequentialSentenceClassification, and MultipleChoice down-stream tasks:
|
37 |
|
38 |
```python
|
39 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
40 |
tokenizer = AutoTokenizer.from_pretrained("kiddothe2b/adhoc-hierarchical-transformer-base-4096", trust_remote_code=True)
|
41 |
+
doc_classifier = AutoModelForSequenceClassification("kiddothe2b/adhoc-hierarchical-transformer-base-4096", trust_remote_code=True)
|
42 |
```
|
43 |
|
44 |
Note: If you wish to use a fully pre-trained HAT model, you have to use [kiddothe2b/hierarchical-transformer-base-4096](https://huggingface.co/kiddothe2b/hierarchical-transformer-base-4096).
|