Update convert_wan.py

convert_wan.py

@@ -9,12 +9,28 @@ import math
 import os
 import gdown
 
-# Add Wan2.1 to path
+# --- 0. FORCE CPU PATCH (CRITICAL FIX) ---
+# The T5 code tries to call torch.cuda.current_device() on import. 
+# We must edit the file on disk BEFORE importing it.
+t5_path = "/app/Wan2.1/wan/modules/t5.py"
+if os.path.exists(t5_path):
+    print("--- PATCHING T5.PY TO REMOVE CUDA REQUIREMENT ---")
+    with open(t5_path, 'r') as f:
+        content = f.read()
+    
+    # Replace the hardcoded cuda call with "cpu"
+    new_content = content.replace("device=torch.cuda.current_device(),", "device='cpu',")
+    
+    with open(t5_path, 'w') as f:
+        f.write(new_content)
+else:
+    print("WARNING: Could not find t5.py to patch!")
+
+# Add Wan2.1 to path AFTER patching
 sys.path.append("/app/Wan2.1")
 
 # --- 1. DOWNLOAD WEIGHTS FROM DRIVE ---
 print("--- DOWNLOADING MODEL FROM GOOGLE DRIVE ---")
-# The ID from your link: 1pf-fhkWayK3_sv9bujvT2Od04ki2Wh4y
 file_id = '1pf-fhkWayK3_sv9bujvT2Od04ki2Wh4y'
 url = f'https://drive.google.com/uc?id={file_id}'
 ckpt_path = "/app/wan13bNSFWFull_v20I2v.safetensors"