Fix Flax LMSDiscreteScheduler sigma indexing and align LMS prediction_type with PyTorch#13556
Open
HaoyuLi-Nova wants to merge 1 commit intohuggingface:mainfrom
Open
Fix Flax LMSDiscreteScheduler sigma indexing and align LMS prediction_type with PyTorch#13556HaoyuLi-Nova wants to merge 1 commit intohuggingface:mainfrom
HaoyuLi-Nova wants to merge 1 commit intohuggingface:mainfrom