fffiloni commited on
Commit
bf3518b
·
verified ·
1 Parent(s): 817f610

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -10
Dockerfile CHANGED
@@ -102,11 +102,8 @@ RUN pip install --no-cache-dir segment-anything==1.0 && \
102
  pip install -e . && \
103
  cd ..
104
 
105
- # Clone the GroundingDINO repository and install it
106
- RUN git clone https://github.com/IDEA-Research/GroundingDINO.git && \
107
- cd GroundingDINO && \
108
- pip install -e . && \
109
- cd ..
110
 
111
  # Create a custom setup.py for GroundingDINO extension
112
  RUN echo "from setuptools import setup\n\
@@ -117,18 +114,19 @@ setup(\n\
117
  ext_modules=[\n\
118
  CUDAExtension(\n\
119
  name='_C',\n\
120
- sources=['/home/user/GroundingDINO/groundingdino/models/GroundingDINO/csrc/ms_deform_attn.cpp', '/home/user/GroundingDINO/groundingdino/models/GroundingDINO/csrc/ms_deform_attn_cuda.cu'],\n\
121
  ),\n\
122
  ],\n\
123
  cmdclass={\n\
124
  'build_ext': BuildExtension\n\
125
  }\n\
126
- )" > /home/user/GroundingDINO/groundingdino/models/GroundingDINO/csrc/setup.py
127
 
128
  # Compile the GroundingDINO custom C++ operations
129
- RUN cd /home/user/GroundingDINO/groundingdino/models/GroundingDINO/csrc && \
130
- python setup.py build_ext --inplace && \
131
- cd /home/user/app
 
132
 
133
  # Upgrade pip and install Gradio
134
  RUN python3 -m pip install --no-cache-dir gradio
 
102
  pip install -e . && \
103
  cd ..
104
 
105
+ # Install GroundingDINO via pip
106
+ RUN pip install --no-cache-dir git+https://github.com/IDEA-Research/GroundingDINO.git
 
 
 
107
 
108
  # Create a custom setup.py for GroundingDINO extension
109
  RUN echo "from setuptools import setup\n\
 
114
  ext_modules=[\n\
115
  CUDAExtension(\n\
116
  name='_C',\n\
117
+ sources=['/home/user/.local/lib/python3.10/site-packages/groundingdino/models/GroundingDINO/csrc/ms_deform_attn.cpp', '/home/user/.local/lib/python3.10/site-packages/groundingdino/models/GroundingDINO/csrc/ms_deform_attn_cuda.cu'],\n\
118
  ),\n\
119
  ],\n\
120
  cmdclass={\n\
121
  'build_ext': BuildExtension\n\
122
  }\n\
123
+ )" > /home/user/setup_groundingdino.py
124
 
125
  # Compile the GroundingDINO custom C++ operations
126
+ RUN cd /home/user && \
127
+ python setup_groundingdino.py build_ext --inplace && \
128
+ rm setup_groundingdino.py
129
+
130
 
131
  # Upgrade pip and install Gradio
132
  RUN python3 -m pip install --no-cache-dir gradio