PR #1096 Flax: avoid recompilation when params change

Flax: avoid recompilation when params change #1096

pcuenca merged 9 commits into main from flax-no-recompile

Do not recompile when guidance_scale changes.

5333895b

Remove debug for simplicity.

e1a7d264

pcuenca marked this pull request as draft 3 years ago

make style

3e9b7535

Merge branch 'main' into flax-no-recompile

16f592c4

Merge remote-tracking branch 'origin/main' into flax-no-recompile

65e07819

Make guidance_scale an array.

759094cd

Merge remote-tracking branch 'origin/main' into flax-no-recompile

7db319bd

pcuenca marked this pull request as ready for review 3 years ago

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca requested a review from

patil-suraj 3 years ago

patil-suraj approved these changes on 2022-12-07

Make DEBUG a constant to avoid passing it down.

bc80eed1

Add comments for clarification.

b55a8a19

pcuenca merged 6a7f1f09 into main 3 years ago

pcuenca deleted the flax-no-recompile branch 3 years ago

Reviewers

patil-suraj

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone