Kohaku-Blueleaf eaa9f5162f Add CPU fp8 support ...

Since norm layer need fp32, I only convert the linear operation layer(conv2d/linear)

And TE have some pytorch function not support bf16 amp in CPU. I add a condition to indicate if the autocast is for unet.

2023-10-24 01:49:05 +08:00

30 KiB

Raw Blame History

View Raw