Fix Wan2.1 - Mode CLI (bypass Gradio)

Date : 18/01/2026 17:30 Status : TERMINE

Probleme initial

Wan2.1 crashait systematiquement pendant la generation video via Gradio API :

• ConnectionResetError apres quelques secondes de generation
• Le process Python mourait sans message d'erreur visible
• Cause racine : Flash Attention 2 non disponible sur Windows

Solution implementee

1. Mode CLI direct

Utilisation de generate.py au lieu de l'interface Gradio :

• Execution via SSH directement sur win11
• Pas de serveur web, pas de connexion persistante
• Le fichier est genere localement puis transfere

2. Patch attention.py

# Avant
from .attention import flash_attention

# Apres (dans I:\Wan2.1\wan\modules\model.py)
from .attention import attention as flash_attention

La fonction attention() a un fallback vers torch.nn.functional.scaled_dot_product_attention quand Flash Attention n'est pas disponible.

3. Wrapper Python avec base64

Pour eviter les problemes d'echappement du prompt dans SSH/CMD :

• Prompt encode en base64
• Script Python temporaire genere sur win11
• Decode et execute le CLI avec les bons parametres

4. SFTP au lieu de SCP

SCP ne supporte pas les chemins Windows correctement. SFTP utilise le format :

/I:/Wan2.1/outputs/file.mp4

Fichiers modifies

1. /stock_8to/33800-stack/projects/ai-orchestrator/app/main.py

   • Nouveau executor execute_wan21_job() avec mode CLI

2. I:\Wan2.1\wan\modules\model.py (sur win11)

   • Import modifie pour utiliser attention avec fallback

Test valide

{
  "job_id": "9cf733b4-5587-46b7-98a7-b0be4a7e3dcf",
  "status": "completed",
  "processing_time_ms": 218744,
  "output_files": ["9cf733b4-5587-46b7-98a7-b0be4a7e3dcf/video.mp4"],
  "size_bytes": 1338019
}

• Prompt: "a colorful sunset over ocean waves"
• Resolution: 480*832
• Steps: 5
• Temps: ~3.5 minutes

Notes

• Le mode Gradio reste disponible via passthrough: true
• Le chargement du modele prend ~2 minutes a chaque job (pas de serveur persistant)
• Pour accelerer, on pourrait implementer un mode "warm" avec serveur Gradio actif

Commits GitLab

• 91d63e3 : Wan2.1: switch to CLI mode (bypass Gradio)
• a98a868 : Fix prompt escaping using base64 wrapper
• 2e52119 : Fix SCP path escaping for Windows
• 670b080 : Use SFTP instead of SCP for Windows file transfer