anas commited on
Commit
da1b9fb
·
1 Parent(s): b7ae41e

Fix checkpoint download: use wget from HF Hub

Browse files
Files changed (1) hide show
  1. 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 huggingface_hub
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
- python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download('AGLO-AI/raster2seq-checkpoints', 'r2g_res256_ep0849.pth', local_dir='checkpoints')"
 
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