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 tests/gnarly_pdfs/test_gnarly_pdfs.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/ \`
moves what happens where around a bit and updates readme 2025-03-13 15:31:13 -07:00			`&& bash tests/gnarly_pdfs/test_gnarly_pdfs.sh`
basic docker image and test 2025-03-13 13:05:09 -07:00