import pathlib volume = modal.Volume.from_name("my-volume") VOL_MOUNT_PATH = pathlib.Path("/vol") @app.function( gpu="A10G", timeout=2 * 60 * 60, # run for at most two hours volumes={VOL_MOUNT_PATH: volume}, ) 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, )