Skip to content

Commit ee90fec

Browse files
committed
fix: use CPU-only PyTorch for Jaguar Space to fit HF free-tier build limits
1 parent 13b0870 commit ee90fec

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

  • spaces/jaguar-reid-megadescriptor-spherical

spaces/jaguar-reid-megadescriptor-spherical/Dockerfile

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,10 @@ RUN pip install --upgrade pip
2323

2424
ARG HYPERVIEW_VERSION=0.3.1
2525

26-
# Install the released PyPI package plus the ML stack needed for timm-image.
27-
RUN pip install "hyperview[ml]==${HYPERVIEW_VERSION}" \
26+
# Install CPU-only PyTorch first (much smaller than default CUDA bundle),
27+
# then the released HyperView package with the ML extras.
28+
RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu \
29+
&& pip install "hyperview[ml]==${HYPERVIEW_VERSION}" \
2830
&& python -c "import hyperview; print('hyperview', hyperview.__version__)"
2931

3032
COPY --chown=user demo.py ./demo.py

0 commit comments

Comments
 (0)