diffusers
155d272c
- Update FlaxLMSDiscreteScheduler (#1474)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
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`
References
#1474 - Update FlaxLMSDiscreteScheduler
Author
dzlab
Parents
2b30b109
Loading