minimax: upgrade torch to cu130 for int8_convrot
Browse files- worker/minimax_start.sh +14 -0
worker/minimax_start.sh
CHANGED
|
@@ -15,6 +15,20 @@ PY="$(command -v python3 || command -v python)"
|
|
| 15 |
|
| 16 |
echo "[minimax] variant=${VARIANT} comfy_target=${TARGET}"
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
# --- 1. ComfyUI version ---------------------------------------------------
|
| 19 |
cd "$COMFY"
|
| 20 |
CUR="$(git describe --tags --exact-match 2>/dev/null || echo unknown)"
|
|
|
|
| 15 |
|
| 16 |
echo "[minimax] variant=${VARIANT} comfy_target=${TARGET}"
|
| 17 |
|
| 18 |
+
# --- 0. torch with CUDA 13 ------------------------------------------------
|
| 19 |
+
# int8_convrot checkpoints need pytorch built against cu130 (per Comfy-Org
|
| 20 |
+
# model card). Base image ships cu128 torch; swap it on first boot.
|
| 21 |
+
if [ "${UPGRADE_TORCH_CU130:-1}" = "1" ]; then
|
| 22 |
+
CUR_CU="$("$PY" -c 'import torch; print(torch.version.cuda or "")' 2>/dev/null || echo '')"
|
| 23 |
+
case "$CUR_CU" in
|
| 24 |
+
13*) echo "[minimax] torch already cu${CUR_CU}" ;;
|
| 25 |
+
*)
|
| 26 |
+
echo "[minimax] upgrading torch cu${CUR_CU:-none} -> cu130"
|
| 27 |
+
"$PY" -m pip install -q -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130
|
| 28 |
+
;;
|
| 29 |
+
esac
|
| 30 |
+
fi
|
| 31 |
+
|
| 32 |
# --- 1. ComfyUI version ---------------------------------------------------
|
| 33 |
cd "$COMFY"
|
| 34 |
CUR="$(git describe --tags --exact-match 2>/dev/null || echo unknown)"
|