examples fix t2i training (#5001)

patrickvonplaten · web-flow · commit 1037287e2bae · 2023-09-12T23:52:41.000+02:00
* examples fix t2i training

* make style
diff --git a/examples/t2i_adapter/train_t2i_adapter_sdxl.py b/examples/t2i_adapter/train_t2i_adapter_sdxl.py
@@ -1060,7 +1060,9 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
     )
 
     # 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)

Original file line number	Diff line number	Diff line change
`@@ -1060,7 +1060,9 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):`
`1060`	`1060`	`)`
`1061`	`1061`
`1062`	`1062`	# Prepare everything with our `accelerator`.
`1063`		`- t2iadapter, optimizer, lr_scheduler = accelerator.prepare(t2iadapter, optimizer, lr_scheduler)`
	`1063`	`+ t2iadapter, optimizer, train_dataloader, lr_scheduler = accelerator.prepare(`
	`1064`	`+ t2iadapter, optimizer, train_dataloader, lr_scheduler`
	`1065`	`+ )`
`1064`	`1066`
`1065`	`1067`	`# We need to recalculate our total training steps as the size of the training dataloader may have changed.`
`1066`	`1068`	`num_update_steps_per_epoch = math.ceil(len(train_dataloader) / args.gradient_accumulation_steps)`