Without importing torch for the float16 versions, it throws an error

Just to keep alphabetical order with the imports, do you mind swapping the order of the two imports?

Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment