diffusers
[Examples] Add streaming support to the ControlNet training example in JAX
#2859

Merged

[Examples] Add streaming support to the ControlNet training example in JAX #2859

sayakpaul merged 10 commits into main from feat/streaming-controlnet

improve stable unclip doc.

a009f1d1

Merge branch 'main' of https://github.com/huggingface/diffusers

ecf008f0

feat: add streaming support to controlnet flax training script.

6550c888

fix: CLI arg.

def534a5

fix: torch dataloader shuffle setting.

4d0ab455

fix: dataset length.

fc8e1ee0

fix: wandb config.

6c82633e

fix: steps_per_epoch in the training loop.

297f2b43

add: entry about streaming in the readme

acc95090

sayakpaul requested a review from

yiyixuxu 2 years ago

sayakpaul requested a review from

patrickvonplaten 2 years ago

patrickvonplaten approved these changes on 2023-03-28

get column names from iterable dataset + fix final logging

142c9ffe

yiyixuxu approved these changes on 2023-03-28

sayakpaul merged d82b0323 into main 2 years ago

sayakpaul deleted the feat/streaming-controlnet branch 2 years ago

Reviewers

yiyixuxu

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone