Spaces:

MogensR
/

VideoBackgroundReplacer2

Paused

MogensR commited on Oct 3, 2025

Commit

66d2cbf

verified ·

1 Parent(s): ec5f699

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -12,7 +12,8 @@ ENV DEBIAN_FRONTEND=noninteractive \
     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6+PTX" \
     CUDA_VISIBLE_DEVICES="0" \
-    OMP_NUM_THREADS=4 \
     OPENBLAS_NUM_THREADS=1 \
     MKL_NUM_THREADS=1 \
     NUMEXPR_NUM_THREADS=1 \

     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6+PTX" \
     CUDA_VISIBLE_DEVICES="0" \
+    OMP_NUM_THREADS=2 \
+RUN export OMP_NUM_THREADS=2
     OPENBLAS_NUM_THREADS=1 \
     MKL_NUM_THREADS=1 \
     NUMEXPR_NUM_THREADS=1 \