Breaking the opacity of language models for legal professionals 📖 Join us by smashing the button at top right 🤗
pip install logfire-callbackexport LOGFIRE_TOKEN=your_logfire_tokenfrom transformers import Trainer, TrainingArguments
from logfire_callback import LogfireCallback
# Initialize your model, dataset, etc.
training_args = TrainingArguments(
output_dir="./results",
num_train_epochs=3,
# ... other training arguments
)
trainer = Trainer(
model=model,
args=training_args,
train_dataset=train_dataset,
callbacks=[LogfireCallback()] # Add the Logfire callback here
)
trainer.train()
