Update train.py
Browse files
train.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer, Trainer, Train
|
|
3 |
from datasets import load_dataset
|
4 |
|
5 |
# Load dataset (Replace this with your dataset)
|
6 |
-
dataset = load_dataset("
|
7 |
|
8 |
# Load model and tokenizer
|
9 |
model_name = "t5-large"
|
|
|
3 |
from datasets import load_dataset
|
4 |
|
5 |
# Load dataset (Replace this with your dataset)
|
6 |
+
dataset = load_dataset("tatsu-lab/alpaca") # Example alternative dataset
|
7 |
|
8 |
# Load model and tokenizer
|
9 |
model_name = "t5-large"
|