Untitled

from transformers import Trainer, TrainingArguments

# Define training arguments
training_args = TrainingArguments(output_dir="./results", num_train_epochs=3, per_device_train_batch_size=8)

# Fine-tune model
trainer = Trainer(model=model, args=training_args, train_dataset=train_dataset, eval_dataset=eval_dataset)
trainer.train()