Skip to content

Commit 1d007d1

Browse files
committed
Perhaps fixing default config
1 parent e7020c7 commit 1d007d1

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

scripts/train/newtrainer-gantry.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,13 +42,12 @@ if ! beaker image create --workspace ai2/oe-data-pdf --name $IMAGE_TAG $IMAGE_TA
4242
fi
4343

4444
gantry run \
45-
--description "${run_name}"\
45+
--description "OlmOCR Training Run"\
4646
--task-name "${run_name}"\
4747
--allow-dirty \
4848
--host-networking \
4949
--workspace ai2/olmocr \
5050
--beaker-image $BEAKER_USER/$IMAGE_TAG \
51-
--pip gantry-train-requirements.txt \
5251
--priority normal \
5352
--gpus 1 \
5453
--preemptible \
@@ -64,4 +63,4 @@ gantry run \
6463
--weka oe-training-default:/weka/oe-training-default \
6564
--shared-memory 10GiB \
6665
--yes \
67-
-- /bin/bash -c "pip install flash-attn==2.8.0.post2 --no-build-isolation && python -m olmocr.train.train --config olmocr/train/configs/qwen25_vl_b100_x1_default_image_1600.yaml"
66+
-- /bin/bash -c "pip install -r gantry-train-requirements.txt && pip install transformers==4.52.4 && pip install flash-attn==2.8.0.post2 --no-build-isolation && /usr/bin/python -m olmocr.train.train --config olmocr/train/configs/qwen25_vl_b100_x1_default.yaml"

0 commit comments

Comments
 (0)