BF16
It's saved as F16 in Full....
as a workaround you can use pipe = pipe.to(torch.bfloat16) to avoid problems during inference
pipe = pipe.to(torch.bfloat16)
· Sign up or log in to comment