From 1037287e2bae30e33f844786f5e65ca5a69c41a1 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 12 Sep 2023 23:52:41 +0200 Subject: [PATCH] examples fix t2i training (#5001) * examples fix t2i training * make style --- examples/t2i_adapter/train_t2i_adapter_sdxl.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/t2i_adapter/train_t2i_adapter_sdxl.py b/examples/t2i_adapter/train_t2i_adapter_sdxl.py index 3d846f42f6..d39081bbfc 100644 --- a/examples/t2i_adapter/train_t2i_adapter_sdxl.py +++ b/examples/t2i_adapter/train_t2i_adapter_sdxl.py @@ -1060,7 +1060,9 @@ def main(args): ) # Prepare everything with our `accelerator`. - t2iadapter, optimizer, lr_scheduler = accelerator.prepare(t2iadapter, optimizer, lr_scheduler) + t2iadapter, optimizer, train_dataloader, lr_scheduler = accelerator.prepare( + t2iadapter, optimizer, train_dataloader, lr_scheduler + ) # We need to recalculate our total training steps as the size of the training dataloader may have changed. num_update_steps_per_epoch = math.ceil(len(train_dataloader) / args.gradient_accumulation_steps)