praeclarumjj3 commited on
Commit
dcaec5d
·
1 Parent(s): a531523
Files changed (1) hide show
  1. Dockerfile +4 -5
Dockerfile CHANGED
@@ -40,13 +40,12 @@ ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
40
 
41
  RUN pip install ninja
42
 
43
- USER root
44
- RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 pip install -e . && cd ..
45
 
46
  RUN sh deform_setup.sh
47
 
48
- # USER user
49
 
50
- # EXPOSE 7860
51
 
52
- # ENTRYPOINT ["python", "gradio_app.py"]
 
40
 
41
  RUN pip install ninja
42
 
43
+ RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 pip install -e . --user && cd ..
 
44
 
45
  RUN sh deform_setup.sh
46
 
47
+ USER user
48
 
49
+ EXPOSE 7860
50
 
51
+ ENTRYPOINT ["python", "gradio_app.py"]