Update src/diffusers/models/transformers/transformer_wan.py

a-r-r-o-w · web-flow · commit 3568e9145510 · 2025-07-29T09:46:55.000+05:30
diff --git a/src/diffusers/models/transformers/transformer_wan.py b/src/diffusers/models/transformers/transformer_wan.py
@@ -70,7 +70,7 @@ class WanAttnProcessor:
     def __init__(self):
         if not hasattr(F, "scaled_dot_product_attention"):
             raise ImportError(
-                "WanAttnProcessor2_0 requires PyTorch 2.0. To use it, please upgrade PyTorch to version 2.0 or higher."
+                "WanAttnProcessor requires PyTorch 2.0. To use it, please upgrade PyTorch to version 2.0 or higher."
             )
 
     def __call__(

Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ class WanAttnProcessor:`
`70`	`70`	`def __init__(self):`
`71`	`71`	`if not hasattr(F, "scaled_dot_product_attention"):`
`72`	`72`	`raise ImportError(`
`73`		`- "WanAttnProcessor2_0 requires PyTorch 2.0. To use it, please upgrade PyTorch to version 2.0 or higher."`
	`73`	`+ "WanAttnProcessor requires PyTorch 2.0. To use it, please upgrade PyTorch to version 2.0 or higher."`
`74`	`74`	`)`
`75`	`75`
`76`	`76`	`def __call__(`