olmocr/scripts/beaker/gpu-ci-script.sh

#/usr/bin/bash

set -ex

git clone https://github.com/allenai/olmocr.git olmocr \
  && cd olmocr \
  && git checkout $GIT_REVISION \
  && /root/.local/bin/uv pip install --system --no-cache \
    .[gpu] \
    pytest \
    --find-links https://flashinfer.ai/whl/cu124/torch2.4/flashinfer/ \
  && bash scripts/run_integration_test.sh
wip 2025-03-13 14:10:55 -07:00			`#/usr/bin/bash`

basic docker image and test 2025-03-13 13:05:09 -07:00			`set -ex`

wip 2025-03-13 14:10:55 -07:00			`git clone https://github.com/allenai/olmocr.git olmocr \`
basic docker image and test 2025-03-13 13:05:09 -07:00			`&& cd olmocr \`
			`&& git checkout $GIT_REVISION \`
wip 2025-03-13 14:10:55 -07:00			`&& /root/.local/bin/uv pip install --system --no-cache \`
			`.[gpu] \`
			`pytest \`
			`--find-links https://flashinfer.ai/whl/cu124/torch2.4/flashinfer/ \`
Trying to get new CI to work 2025-03-14 02:43:55 +00:00			`&& bash scripts/run_integration_test.sh`
basic docker image and test 2025-03-13 13:05:09 -07:00