Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 16

Commit

4bd86a1

•

1 Parent(s): 133c73d

Update build_ops.sh

Files changed (1) hide show

build_ops.sh CHANGED Viewed

@@ -18,5 +18,9 @@ export CC=/usr/bin/gcc-11 # this ensures that gcc 11 is being used for compilati
 #echo $CUDA_HOME
 cd ./models/GroundingDINO/ops
 python ./setup.py build install
 python ./test.py # should result in 6 lines of * True
-cd ../../../

 #echo $CUDA_HOME
 cd ./models/GroundingDINO/ops
 python ./setup.py build install
+wheel convert ./dist/MultiScaleDeformableAttention-1.0-py3.10-linux-x86_64.egg
+mv MultiScaleDeformableAttention-1.0-py310-cp310-linux_x86_64.whl MultiScaleDeformableAttention-1.0-cp310-cp310-linux_x86_64.whl
+pip install MultiScaleDeformableAttention-1.0-cp310-cp310-linux_x86_64.whl
 python ./test.py # should result in 6 lines of * True
+cd ../../../
+cp ./models/GroundingDINO/ops/build/lib.linux-x86_64-cpython-310/MultiScaleDeformableAttention.cpython-310-x86_64-linux-gnu.so .