File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -42,13 +42,12 @@ if ! beaker image create --workspace ai2/oe-data-pdf --name $IMAGE_TAG $IMAGE_TA
42
42
fi
43
43
44
44
gantry run \
45
- --description " ${run_name} " \
45
+ --description " OlmOCR Training Run " \
46
46
--task-name " ${run_name} " \
47
47
--allow-dirty \
48
48
--host-networking \
49
49
--workspace ai2/olmocr \
50
50
--beaker-image $BEAKER_USER /$IMAGE_TAG \
51
- --pip gantry-train-requirements.txt \
52
51
--priority normal \
53
52
--gpus 1 \
54
53
--preemptible \
@@ -64,4 +63,4 @@ gantry run \
64
63
--weka oe-training-default:/weka/oe-training-default \
65
64
--shared-memory 10GiB \
66
65
--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"
You can’t perform that action at this time.
0 commit comments