Bf16 not supported

by samuelqy - opened Jan 16, 2024

Jan 16, 2024

We only have v100s which don’t support bf16. Could you release f16/f32 versions?

Beijing Academy of Artificial Intelligence org Jan 17, 2024

you can set the torch_dtype to torch.float16

samuelqy changed discussion status to closed Jan 17, 2024

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment