Update wanForProduction.sh
Browse files- 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-
|
| 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
|