rskuzma's picture
add torch_dtype="auto" to load model weights in bf16
a9e1351