Update README.md
Browse files
README.md
CHANGED
|
@@ -12,8 +12,6 @@ pipeline_tag: text-classification
|
|
| 12 |
|
| 13 |
<!-- PERT adapter for sentiment-classification. -->
|
| 14 |
|
| 15 |
-
|
| 16 |
-
|
| 17 |
## Usage
|
| 18 |
|
| 19 |
First, install `transformers` and `peft`:
|
|
@@ -30,7 +28,6 @@ from transformers import AutoModelForSequenceClassification, RobertaConfig
|
|
| 30 |
|
| 31 |
config = RobertaConfig.from_pretrained(
|
| 32 |
"roberta-base",
|
| 33 |
-
num_labels=2,
|
| 34 |
id2label={ 0: "👎", 1: "👍"}
|
| 35 |
)
|
| 36 |
model = AutoModelForSequenceClassification.from_pretrained("roberta-base", config=config)
|
|
|
|
| 12 |
|
| 13 |
<!-- PERT adapter for sentiment-classification. -->
|
| 14 |
|
|
|
|
|
|
|
| 15 |
## Usage
|
| 16 |
|
| 17 |
First, install `transformers` and `peft`:
|
|
|
|
| 28 |
|
| 29 |
config = RobertaConfig.from_pretrained(
|
| 30 |
"roberta-base",
|
|
|
|
| 31 |
id2label={ 0: "👎", 1: "👍"}
|
| 32 |
)
|
| 33 |
model = AutoModelForSequenceClassification.from_pretrained("roberta-base", config=config)
|