Commit
•
c5747c7
1
Parent(s):
2b38199
Update README.md
Browse files
README.md
CHANGED
@@ -25,9 +25,9 @@ This repository contains a version of [TeenyTinyLlama-162m]() fine-tuned on a tr
|
|
25 |
|
26 |
import evaluate
|
27 |
import numpy as np
|
28 |
-
from datasets import load_dataset
|
29 |
from transformers import AutoTokenizer, DataCollatorWithPadding
|
30 |
-
from transformers import AutoModelForSequenceClassification, TrainingArguments,
|
31 |
|
32 |
# Load the task
|
33 |
dataset = load_dataset("christykoh/imdb_pt")
|
|
|
25 |
|
26 |
import evaluate
|
27 |
import numpy as np
|
28 |
+
from datasets import load_dataset
|
29 |
from transformers import AutoTokenizer, DataCollatorWithPadding
|
30 |
+
from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer
|
31 |
|
32 |
# Load the task
|
33 |
dataset = load_dataset("christykoh/imdb_pt")
|