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

Remove diff_ras compilation (only needed for training, not inference)

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -17,10 +17,9 @@ 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
 
17
  RUN pip3 install gradio
18
 
19
  RUN cd models/ops && python3 setup.py build install && cd ../..
 
20
 
21
  RUN mkdir -p checkpoints && \
22
+ wget -q -O checkpoints/r2g_res256_ep0849.pth \
23
  "https://huggingface.co/AGLO-AI/raster2seq-checkpoints/resolve/main/r2g_res256_ep0849.pth"
24
 
25
  RUN chown -R user:user /app