Spaces:
Sleeping
Sleeping
| import pathlib | |
| volume = modal.Volume.from_name("my-volume") | |
| VOL_MOUNT_PATH = pathlib.Path("/vol") | |
| def finetune(): | |
| from transformers import Seq2SeqTrainer | |
| ... | |
| training_args = Seq2SeqTrainingArguments( | |
| output_dir=str(VOL_MOUNT_PATH / "model"), | |
| # ... more args here | |
| ) | |
| trainer = Seq2SeqTrainer( | |
| model=model, | |
| args=training_args, | |
| train_dataset=tokenized_xsum_train, | |
| eval_dataset=tokenized_xsum_test, | |
| ) |