Update wan/modules/attention.py

wan/modules/attention.py

@@ -2,22 +2,31 @@
 import torch
 
 try:
+    print("calling import flash_attn_interface")
     import flash_attn_interface
 
     def is_hopper_gpu():
+        print("is_hopper_gpu(): checking if not torch.cuda.is_available()")
         if not torch.cuda.is_available():
+            print("is_hopper_gpu(): turch.cuda is not available, so this is not Hopper GPU")
             return False
         device_name = torch.cuda.get_device_name(0).lower()
+        print(f"is_hopper_gpu(): device_name = {device_name}")
         return "h100" in device_name or "hopper" in device_name
     FLASH_ATTN_3_AVAILABLE = is_hopper_gpu()
-except ModuleNotFoundError:
+except ModuleNotFoundError as e:
+    print(f"Got a ModuleNotFoundError for Flash Attention 3: {e}")
     FLASH_ATTN_3_AVAILABLE = False
+print(f"FLASH_ATTN_3_AVAILABLE ? -> {FLASH_ATTN_3_AVAILABLE}")
 
 try:
+    print("calling import flash_attn")
     import flash_attn
     FLASH_ATTN_2_AVAILABLE = True
-except ModuleNotFoundError:
+except ModuleNotFoundError as e:
+    print(f"Got a ModuleNotFoundError for Flash Attention 2: {e}")
     FLASH_ATTN_2_AVAILABLE = False
+print(f"FLASH_ATTN_2_AVAILABLE ? -> {FLASH_ATTN_2_AVAILABLE}")
 
 # FLASH_ATTN_3_AVAILABLE = False