PR #1765 [Dreambooth] flax fixes

[Dreambooth] flax fixes #1765

pcuenca merged 3 commits into main from dreambooth-flax-fixes

Fail if there are less images than the effective batch size.

01ccc9a9

Remove lr-scheduler arg as it's currently ignored.

08ec28fe

Make guidance_scale work for batch_size > 1.

32d6d8d9

pcuenca requested a review from

patil-suraj 3 years ago

pcuenca requested a review from

yiyixuxu 3 years ago

pcuenca requested a review from

patrickvonplaten 3 years ago

yiyixuxu approved these changes on 2022-12-19

patrickvonplaten approved these changes on 2022-12-19

pcuenca merged 9f8c915a into main 3 years ago

pcuenca deleted the dreambooth-flax-fixes branch 3 years ago

Reviewers

yiyixuxu

patrickvonplaten

patil-suraj

Assignees

No one assigned

Labels

None yet

Milestone

No milestone