adbrasi commited on
Commit
c8c7a01
·
verified ·
1 Parent(s): a909e08

Update wanForProduction.sh

Browse files
Files changed (1) hide show
  1. wanForProduction.sh +2 -2
wanForProduction.sh CHANGED
@@ -460,7 +460,7 @@ log_success "ComfyUI pronto"
460
 
461
  # 3. Instalar PyTorch (CUDA 12.8)
462
  log_info "[3/6] Instalando PyTorch CUDA 12.8..."
463
- python3 -m pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128 || log_error "Falha ao instalar PyTorch"
464
  log_success "PyTorch configurado"
465
 
466
  # 4. Instalar SageAttention (condicional)
@@ -470,7 +470,7 @@ import torch, sys
470
  sys.exit(0 if torch.cuda.is_available() else 1)
471
  PY
472
  then
473
- if python3 -m pip install "https://huggingface.co/adbrasi/comfywheel/resolve/main/sageattention-2.2.0-cp311-cp311-linux_x86_64.whl" -q; then
474
  if python3 - <<'PY' >/dev/null 2>&1
475
  import sageattention
476
  PY
 
460
 
461
  # 3. Instalar PyTorch (CUDA 12.8)
462
  log_info "[3/6] Instalando PyTorch CUDA 12.8..."
463
+ python3 -m pip install --force --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128 || log_error "Falha ao instalar PyTorch"
464
  log_success "PyTorch configurado"
465
 
466
  # 4. Instalar SageAttention (condicional)
 
470
  sys.exit(0 if torch.cuda.is_available() else 1)
471
  PY
472
  then
473
+ if python3 -m pip install "https://huggingface.co/adbrasi/comfywheel/resolve/main/sageattention-2.2.0-cp312-cp312-linux_x86_64.whl" -q; then
474
  if python3 - <<'PY' >/dev/null 2>&1
475
  import sageattention
476
  PY