pcuenq's picture
pcuenq HF staff
Fix dtype to work in bfloat16, float16, float32
0cdd7da