diff --git a/scripts/train/newtrainer-gantry.sh b/scripts/train/newtrainer-gantry.sh index 93db3df..f1e3499 100755 --- a/scripts/train/newtrainer-gantry.sh +++ b/scripts/train/newtrainer-gantry.sh @@ -60,6 +60,8 @@ gantry run \ --env-secret AWS_ACCESS_KEY_ID=S2_AWS_ACCESS_KEY_ID \ --env-secret AWS_SECRET_ACCESS_KEY=S2_AWS_SECRET_ACCESS_KEY \ --env-secret WANDB_API_KEY=JAKE_WANDB_API_KEY \ + --weka oe-data-default:/weka/oe-data-default \ + --weka oe-training-default:/weka/oe-training-default \ --shared-memory 10GiB \ --yes \ -- /bin/bash -c "source scripts/beaker/jupiter-ib.sh && python -m olmocr.train.train --config olmocr/train/configs/example_config.yaml" \ No newline at end of file