Bug Fix: Remove explicit message argument in deprecate (#2421)
Remove explicit message argument
This commit is contained in:
parent
5e5ce13e2f
commit
e75eae3711
|
@ -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."
|
||||
|
|
Loading…
Reference in New Issue