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()