yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-12-06 12:11:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Final uv fixes

Browse Source
This commit is contained in:
Jake Poznanski 2025-06-02 18:39:32 +00:00
parent 97da87a3b2
commit 153f1e58b7
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@ -49,6 +49,9 @@ WORKDIR /root
COPY pyproject.toml pyproject.toml COPY pyproject.toml pyproject.toml
COPY olmocr/version.py olmocr/version.py COPY olmocr/version.py olmocr/version.py
# Needed to resolve setuptools dependencies
ENV UV_INDEX_STRATEGY="unsafe-best-match"
RUN uv pip install --system --no-cache -e . RUN uv pip install --system --no-cache -e .
RUN uv pip install --system --no-cache ".[gpu]" --extra-index-url https://download.pytorch.org/whl/cu128 RUN uv pip install --system --no-cache ".[gpu]" --extra-index-url https://download.pytorch.org/whl/cu128
RUN uv pip install --system --no-cache ".[bench]" RUN uv pip install --system --no-cache ".[bench]"
@ -57,5 +60,4 @@ RUN playwright install chromium
COPY olmocr olmocr COPY olmocr olmocr
COPY scripts scripts COPY scripts scripts
RUN python3 -m sglang.launch_server --help
RUN python3 -m olmocr.pipeline --help RUN python3 -m olmocr.pipeline --help