update Docker

Dockerfile

@@ -55,6 +55,9 @@ RUN pip install \
     torchaudio==2.3.0 \
     --index-url https://download.pytorch.org/whl/cu121
 
+
+# Tell PyTorch which CUDA archs to compile for (T4, A100, etc.)
+ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
 # -----------------------------------------------------------------------------
 # Copy requirements & install Python deps
 # (We install requirements AFTER torch so pip doesn't upgrade it to 2.6+)
@@ -83,10 +86,9 @@ RUN pip install --no-build-isolation .
 # (adjust the path if your repo structure is slightly different)
 # WORKDIR /app/VG/groundingdino/models/GroundingDINO
 # RUN if [ -f setup.py ]; then python setup.py build_ext --inplace; fi
-ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
 
 # Build the CUDA ops explicitly
-WORKDIR /app/VG/groundingdino/models/ops
+WORKDIR /app/VG
 RUN python setup.py build_ext --inplace && python setup.py install
 
 # Back to app root