Merge pull request #1327 from bghira/bugfix/sdxl-time-ids

fix SDXL time ids
bghira · Feb 22, 2025 · 23578aa · 23578aa
2 parents a98bb83 + 918fe46
commit 23578aa
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/helpers/training/trainer.py b/helpers/training/trainer.py
@@ -2362,9 +2362,7 @@ def prepare_batch(self, batch: list):
                 **target_device_kwargs
             )
         if time_ids is not None and hasattr(time_ids, "to"):
-            batch["added_cond_kwargs"]["time_ids"] = (
-                time_ids.to(**target_device_kwargs),
-            )
+            batch["added_cond_kwargs"]["time_ids"] = time_ids.to(**target_device_kwargs)
 
         latents = batch.get("latent_batch")
         if not hasattr(latents, "to"):