Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
@@ -115,6 +115,7 @@ setup(\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\
|
|
|
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 |
+
extra_compile_args={'cxx': ['-g'], 'nvcc': ['-O2', '-arch=sm_70']},\n\
|
119 |
),\n\
|
120 |
],\n\
|
121 |
cmdclass={\n\
|