temporary revert to CUDA 12.6

Signed-off-by: bigcat88 <[email protected]>

Author: bigcat88

Dockerfile

@@ -59,7 +59,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     elif [ "$BUILD_TYPE" = "cpu" ]; then \
         venv/bin/python -m pip install torch==2.7.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu; \
     else \
-        venv/bin/python -m pip install torch==2.7.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu128; \
+        venv/bin/python -m pip install torch==2.7.0 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126; \
     fi
 
 COPY ex_app/lib/exclude_nodes.py ex_app/lib/exclude_flows.py /ex_app/lib/