`scaled_dot_product_attention` error
#119
by
rihao-he
- opened
I met an error,
attention.py", line 124, in init
raise ValueError(
ValueError: The current PyTorch version does not support the scaled_dot_product_attention function.
which version should I use?
To use scaled_dot_product_attention for PyTorch, you need PyTorch version >= 2.0