Input type (float) and bias type (c10::BFloat16) should be the same

[fallbernana123456]

由于不支持bfloat16 ,我 把 b, c 改成stage_b_lite.safetensors , stage_c.safetensors ,dtype 改成float32 ,但是会上面这个错误。这个该怎么修改？