PR #690 Flax: add shape argument to `set_timesteps`

Flax: add shape argument to `set_timesteps` #690

pcuenca merged 2 commits into main from fix-flax-schedulers

Flax: add shape argument to set_timesteps

58d03eba

pcuenca requested a review from

patil-suraj 3 years ago

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca requested a review from

kashif 3 years ago

style

f576cc68

patrickvonplaten approved these changes on 2022-10-03

patil-suraj approved these changes on 2022-10-03

pcuenca merged 249b36cc into main 3 years ago

pcuenca deleted the fix-flax-schedulers branch 3 years ago

Reviewers

patil-suraj

patrickvonplaten

kashif

Assignees

No one assigned

Labels

None yet

Milestone

No milestone