search

pixeli99 / SVD_Xtend

Stable Video Diffusion Training Code and Extensions.
481 stars 45 forks source link

enable autocast for bf16 as well #44

Open furkan-celik opened 4 months ago

furkan-celik commented 4 months ago

In my experiments with A100, I have seen type mismatch errors when bf16 is selected. Which I believe is a bug and this simple change fixed it for me.