Bug Fix: Remove explicit message argument in deprecate (#2421)

Remove explicit message argument

Bug Fix: Remove explicit message argument in deprecate (#2421)
Remove explicit message argument
e75eae37 · alvanli · GitHub · 5e5ce13e · e75eae37
Unverified Commit e75eae37 authored Mar 03, 2023 by alvanli Committed by GitHub Mar 03, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/models/cross_attention.py src/diffusers/models/cross_attention.py +1 -1

No files found.
--- a/src/diffusers/models/cross_attention.py
+++ b/src/diffusers/models/cross_attention.py
@@ -260,7 +260,7 @@ class CrossAttention(nn.Module):
            deprecate(
                "batch_size=None",
                "0.0.15",
-                message=(
+                (
                    "Not passing the `batch_size` parameter to `prepare_attention_mask` can lead to incorrect"
                    " attention mask preparation and is deprecated behavior. Please make sure to pass `batch_size` to"
                    " `prepare_attention_mask` when preparing the attention_mask."