Update src/diffusers/models/attention_flax.py

sayakpaul · DN6 · web-flow · commit 0605c836f750 · 2025-08-22T22:00:26.000+05:30
Co-authored-by: Dhruv Nair &lt;dhruv.nair@gmail.com&gt;
diff --git a/src/diffusers/models/attention_flax.py b/src/diffusers/models/attention_flax.py
@@ -157,7 +157,7 @@ class FlaxAttention(nn.Module):
 
     def setup(self):
         logger.warning(
-            "Flax classes are deprecated and will be removed in Diffusers v1. We "
+            "Flax classes are deprecated and will be removed in Diffusers v1.0.0 We "
             "recommend migrating to PyTorch classes or pinning your version of Diffusers."
         )
 

Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,7 @@ class FlaxAttention(nn.Module):`
`157`	`157`
`158`	`158`	`def setup(self):`
`159`	`159`	`logger.warning(`
`160`		`- "Flax classes are deprecated and will be removed in Diffusers v1. We "`
	`160`	`+ "Flax classes are deprecated and will be removed in Diffusers v1.0.0 We "`
`161`	`161`	`"recommend migrating to PyTorch classes or pinning your version of Diffusers."`
`162`	`162`	`)`
`163`	`163`