Bit-Gpt-0-2-7 / app.py
Piux24's picture
Update app.py
beddc21 verified
raw
history blame contribute delete
523 Bytes
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()