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