Cómo instalar ComfyUI con Krea 2 Turbo INT8 en Google Colab: guía paso a paso (gratis y en la nube)

# # ==============================================================================
# COMFYUI + KREA 2 TURBO INT8 (WORKFLOW T2I) - GOOGLE COLAB
# ==============================================================================
import os
import shutil
import time
import threading
import getpass

# ==============================================================================
# 1. CONFIGURACIÓN DEL DIRECTORIO
# ==============================================================================
%cd /content
COMFY_PATH = "/content/ComfyUI"

# Elimina instalación anterior para garantizar un entorno limpio
if os.path.exists(COMFY_PATH):
    shutil.rmtree(COMFY_PATH)

# ==============================================================================
# 2. CLONAR COMFYUI
# ==============================================================================
print("📥 Clonando ComfyUI (versión más reciente con soporte nativo para Krea-2)...")
!git clone --depth 1 https://github.com/comfyanonymous/ComfyUI.git
%cd {COMFY_PATH}

# ==============================================================================
# 3. DEPENDENCIAS
# ==============================================================================
print("📦 Instalando dependencias de ComfyUI y Ngrok...")
!pip install -q -r requirements.txt
!pip install -q pyngrok

# ==============================================================================
# 4. ESTRUCTURA DE CARPETAS
# ==============================================================================
os.makedirs("models/diffusion_models", exist_ok=True)
os.makedirs("models/text_encoders", exist_ok=True)
os.makedirs("models/vae", exist_ok=True)
os.makedirs("models/loras", exist_ok=True)

# ==============================================================================
# 5. DESCARGA DE LOS MODELOS (SOLO LOS NECESARIOS PARA EL WORKFLOW)
# ==============================================================================

print("\n📥 Descargando modelos de Krea 2 Turbo INT8...")

print("  ├─ Diffusion Model (krea2_turbo_int8_convrot - ~13.8GB)...")
!wget -q --show-progress \
    https://huggingface.co/Comfy-Org/Krea-2/resolve/main/diffusion_models/krea2_turbo_int8_convrot.safetensors \
    -O models/diffusion_models/krea2_turbo_int8_convrot.safetensors

print("  ├─ Text Encoder (qwen3vl_4b_fp8_scaled)...")
!wget -q --show-progress \
    https://huggingface.co/Comfy-Org/Krea-2/resolve/main/text_encoders/qwen3vl_4b_fp8_scaled.safetensors \
    -O models/text_encoders/qwen3vl_4b_fp8_scaled.safetensors

print("  ├─ VAE (qwen_image_vae)...")
!wget -q --show-progress \
    https://huggingface.co/Comfy-Org/Krea-2/resolve/main/vae/qwen_image_vae.safetensors \
    -O models/vae/qwen_image_vae.safetensors

print("  └─ LoRA de ejemplo (krea2_darkbrush - para el workflow)...")
!wget -q --show-progress \
    https://huggingface.co/Comfy-Org/Krea-2/resolve/main/loras/krea2_darkbrush.safetensors \
    -O models/loras/krea2_darkbrush.safetensors

# ==============================================================================
# 6. LIMPIEZA DE CACHÉ (PARA AHORRAR ESPACIO EN EL DISCO DE COLAB)
# ==============================================================================
print("\n🧹 Limpiando caché de instalación...")
!rm -rf /root/.cache/pip
!rm -rf /tmp/*
!find . -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null

print("\n✅ ¡COMFYUI + KREA 2 TURBO INT8 INSTALADOS CON ÉXITO!")

# ==============================================================================
# 7. INICIAR COMFYUI EN SEGUNDO PLANO
# ==============================================================================
def start_comfy():
    os.system("python main.py --listen 0.0.0.0 --port 8188")

print("🚀 Iniciando ComfyUI en segundo plano...")
threading.Thread(target=start_comfy, daemon=True).start()

# Espera a que el servidor se inicialice completamente
time.sleep(12)

# ==============================================================================
# 8. CONFIGURACIÓN NGROK (TÚNEL SEGURO Y ESTABLE)
# ==============================================================================
from pyngrok import ngrok

print("\n🔑 Consigue tu token gratuito en: https://dashboard.ngrok.com/get-started/your-authtoken")
NGROK_TOKEN = getpass.getpass("Pega aquí el token de Ngrok: ")

ngrok.kill()
ngrok.set_auth_token(NGROK_TOKEN)

# Crea el túnel en el puerto 8188
public_url = ngrok.connect(8188)

print("\n" + "="*70)
print(f"🎉 ¡COMFYUI + KREA 2 TURBO INT8 EN LÍNEA!")
print(f"🔗 Accede a tu enlace seguro: {public_url}")
print("="*70)
print("\n💡 PRÓXIMOS PASOS:")
print("1. Arrastra el archivo 'image_krea2_turbo_t2i_int8.json' a la interfaz.")
print("2. Todos los modelos ya están en las carpetas correctas y serán reconocidos.")
print("3. Para agregar más LoRAs de estilo, descárgalos de:")
print("   https://huggingface.co/Comfy-Org/Krea-2/tree/main/loras")
print("\n📁 ESTRUCTURA DE MODELOS INSTALADOS:")
print("  models/")
print("  ├── diffusion_models/")
print("  │   └── krea2_turbo_int8_convrot.safetensors")
print("  ├── text_encoders/")
print("  │   └── qwen3vl_4b_fp8_scaled.safetensors")
print("  ├── vae/")
print("  │   └── qwen_image_vae.safetensors")
print("  └── loras/")
print("      └── krea2_darkbrush.safetensors")