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/ \
  && pytest tests/gnarly_pdfs/test_evaluate_gnarly_pdfs.py
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/ \`
			`&& pytest tests/gnarly_pdfs/test_evaluate_gnarly_pdfs.py`
basic docker image and test 2025-03-13 13:05:09 -07:00