TypeError: __init__() 收到一个意外的关键字参数 'qk_scale'

我们运行代码加载模型，并收到以下错误： File "/mlx_devbox/users/liuchang.lab/repo/22056/mae/inference.py", line 96, in <module> model_mae = prepare_model(chkpt_dir, 'mae_vit_large_patch16') File "/mlx_devbox/users/liuchang.lab/repo/22056/mae/inference.py", line 29, in prepare_model model = getattr(models_mae, arch)() File "/mlx_devbox/users/liuchang.lab/repo/22056/mae/models_mae.py", line 232, in mae_vit_large_patch16_dec512d8b model = MaskedAutoencoderViT( File "/mlx_devbox/users/liuchang.lab/repo/22056/mae/models_mae.py", line 39, in __init__ self.blocks = nn.ModuleList([ File "/mlx_devbox/users/liuchang.lab/repo/22056/mae/models_mae.py", line 40, in <listcomp> Block(embed_dim, num_heads, mlp_ratio, qkv_bias=True, qk_scale=None, norm_layer=norm_layer)