nikdevs commited on
Commit
b7d4604
·
verified ·
1 Parent(s): b66c0ec

minimax: upgrade torch to cu130 for int8_convrot

Browse files
Files changed (1) hide show
  1. 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)"