Spaces:
Runtime error
Runtime error
File size: 523 Bytes
beddc21 8531778 cd80ce3 8531778 cd80ce3 8531778 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | import os
os.system("pip install transformers datasets torch gradio") # Install missing libraries
from transformers import Trainer, TrainingArguments
# Rest of your code...
from transformers import Trainer, TrainingArguments
training_args = TrainingArguments(
output_dir="./results",
num_train_epochs=3,
per_device_train_batch_size=8,
save_steps=10_000,
save_total_limit=2,
)
trainer = Trainer(
model=model,
args=training_args,
train_dataset="training_data.jsonl",
)
trainer.train()
|