Make FlaxLMSDiscreteScheduler jittable (#2180)#8

srlynch1 wants to merge 2 commits into

mainsrlynch1/diffusers:mainfrom

e2e/diffusers-2180srlynch1/diffusers:e2e/diffusers-2180Copy head branch name to clipboard