diffusers
155d272c - Update FlaxLMSDiscreteScheduler (#1474)

Commit
3 years ago
Update FlaxLMSDiscreteScheduler (#1474) - 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`
Author
Parents
Loading