Hugging Face's logo

Spaces:
shi-labs
/
OneFormer
Runtime error

App Files Community
11
praeclarumjj3 commited on
Commit
a531523
·
1 Parent(s): 0dd3195

Fix user

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -41,7 +41,7 @@ ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
41
  RUN pip install ninja
42
 
43
  USER root
44
- RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && pip install -e . && cd ..
45
 
46
  RUN sh deform_setup.sh
47
 
 
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