TypeError: scaled_dot_product_attention(): argument 'is_causal' must be bool, not Tensor

[AzeeloveJiaq]

No description provided.

[csukuangfj]

please describe your issue in detail.

[leoliu1221]

I guess your issue is when converting to onnx, it has that error.
I solved it by changing whisper/model.py qkv_attention to use n_ctx = int(n_ctx) cause sometimes n_ctx has the type <class 'torch.Tensor'>

[csukuangfj]

If he/she uses a recent version of pytorch, e.g., torch 2.6.0, it should also work.

But without any further information, we cannot say much.

[csukuangfj]

Closing since no reply is received.