Spaces:
Runtime error
Runtime error
anas commited on
Commit ·
da1b9fb
1
Parent(s): b7ae41e
Fix checkpoint download: use wget from HF Hub
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
|
@@ -14,13 +14,14 @@ ENV FORCE_CUDA=1
|
|
| 14 |
RUN pip3 install torch==2.3.1 torchvision==0.18.1 \
|
| 15 |
--index-url https://download.pytorch.org/whl/cu118
|
| 16 |
RUN pip3 install -r requirements.txt
|
| 17 |
-
RUN pip3 install gradio
|
| 18 |
|
| 19 |
RUN cd models/ops && python3 setup.py build install && cd ../..
|
| 20 |
RUN cd diff_ras && python3 setup.py build install && cd ..
|
| 21 |
|
| 22 |
RUN mkdir -p checkpoints && \
|
| 23 |
-
|
|
|
|
| 24 |
|
| 25 |
RUN chown -R user:user /app
|
| 26 |
USER user
|
|
|
|
| 14 |
RUN pip3 install torch==2.3.1 torchvision==0.18.1 \
|
| 15 |
--index-url https://download.pytorch.org/whl/cu118
|
| 16 |
RUN pip3 install -r requirements.txt
|
| 17 |
+
RUN pip3 install gradio
|
| 18 |
|
| 19 |
RUN cd models/ops && python3 setup.py build install && cd ../..
|
| 20 |
RUN cd diff_ras && python3 setup.py build install && cd ..
|
| 21 |
|
| 22 |
RUN mkdir -p checkpoints && \
|
| 23 |
+
wget -q --show-progress -O checkpoints/r2g_res256_ep0849.pth \
|
| 24 |
+
"https://huggingface.co/AGLO-AI/raster2seq-checkpoints/resolve/main/r2g_res256_ep0849.pth"
|
| 25 |
|
| 26 |
RUN chown -R user:user /app
|
| 27 |
USER user
|