无法输出output，麻烦大佬救救！

[Jiegenglyu]

在使用实例代码后会报错无法输出output
from fightingcv_attention.attention.MobileViTv2Attention import *

if name == 'main':
input=torch.randn(50,49,512)
sa = MobileViTv2Attention(d_model=512)
output=sa(input)
print(output.shape)
Launching pytest with arguments

============================= test session starts =============================
collecting ... collected 0 items

============================ no tests ran in 0.74s ============================

Process finished with exit code 5

Empty suite
有没有大佬知道是为什么