运行报错

[seraph667794]

UserWarning: where received a uint8 condition tensor. This behavior is deprecated and will be removed in a future version of PyTorch. Use a boolean condition instead. (Triggered internally at ../aten/src/ATen/native/TensorCompare.cpp:402.) attn_weights = torch.where(causal_mask, attn_weights, mask_value) UserWarning: where received a uint8 condition tensor. This behavior is deprecated and will be removed in a future version of PyTorch. Use a boolean condition instead. (Triggered internally at ../aten/src/ATen/native/TensorCompare.cpp:402.) attn_weights = torch.where(causal_mask, attn_weights, mask_value)

[Hzfinfdu]

这是个warning，应该不会影响实际效果