shuailong
shuailong

> @shuailong616 Hi! Can you try swapping the `dtype=torch.half` to `dtype=torch.bfloat16`? thank you so much for your replay , use torch.bfloat16 have a currect result.
![Uploading iwEeAqNwbmcDAQTRBN0F0QDwBrCTfCCdbk_4pQbKkrYcZP4AB9ITCQwDCAAJomltCgAL0gACCvw.png_720x720q90.jpg…]()
> @shuailong616 Hi! Can you try swapping the `dtype=torch.half` to `dtype=torch.bfloat16`? Hi! I also want to konw why torch.half result wrong reason . looking fowward to your reply