Update FlaxLMSDiscreteScheduler

[dzlab]

• Add the missing scale_model_input method to FlaxLMSDiscreteScheduler
• Use jnp.append for appending to state.derivatives
• Use jnp.delete to pop from state.derivatives

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[patrickvonplaten]

Hey @dzlab,

Could you run make fix-copies to fix the failing test? :-)

[dzlab]

@patrickvonplaten thanks for reviewing, I tried running make fix-copies but this does not seem to perform any changes that could be committed. This is what I tried:

$ cd diffusers
$ git checkout patch-1
Branch 'patch-1' set up to track remote branch 'patch-1' from 'origin'.
$ make fix-copies
python utils/check_copies.py --fix_and_overwrite
python utils/check_dummies.py --fix_and_overwrite
$ git status
On branch patch-1
Your branch is up to date with 'origin/patch-1'.

nothing to commit, working tree clean

[patrickvonplaten]

Fixed it on "main" - thanks for the PR!