mmoskal's picture
Set torch_dtype to bf16
81a9a92