any error in it?
.to(torch.bfloat16) on both model and input to the model
· Sign up or log in to comment