Fix Ideogram4MRoPE collapsing under torch.autocast (compute rotary in float32)#13922
Open
HaozheZhang6 wants to merge 12 commits into
Open
Fix Ideogram4MRoPE collapsing under torch.autocast (compute rotary in float32)#13922HaozheZhang6 wants to merge 12 commits into
Ideogram4MRoPE collapsing under torch.autocast (compute rotary in float32)#13922HaozheZhang6 wants to merge 12 commits into
background
wait
wait-all
cancel
parallel
Loading