PR #2818 [WIP]Flax training script for controlnet

[WIP]Flax training script for controlnet #2818

yiyixuxu merged 12 commits into main from train-controlnet-flax

add train_controlnet_flax

1a335644

fix

b5f71ddf

fix

075f4c4a

improve logging

6c6718c5

gradient_accumulation_steps

63bef8d7

fix

d9cf213e

yiyixuxu changed the title ~~[WIF]Flax training script for controlnet~~ [WIP]Flax training script for controlnet 2 years ago

patrickvonplaten commented on 2023-03-27

patrickvonplaten approved these changes on 2023-03-27

Apply suggestions from code review

bdc71d62

save_model_card

ad4137b9

readme

162697ce

fix

64a9c020

fix

5eec46ed

fix

edd8d76e

yiyixuxu merged d4f846fa into main 2 years ago

yiyixuxu deleted the train-controlnet-flax branch 2 years ago

Reviewers

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone