From 8d09eb44ec19d7b5a535874120811e3d50c265e5 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 15 Aug 2023 18:02:14 -0600 Subject: [PATCH] Fixed an issue with CFG time embeds on SDXL --- toolkit/stable_diffusion_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py index 01533d41..dcdb2638 100644 --- a/toolkit/stable_diffusion_model.py +++ b/toolkit/stable_diffusion_model.py @@ -444,8 +444,8 @@ class StableDiffusion: if do_classifier_free_guidance: # todo check this with larget batches - train_util.concat_embeddings( - add_time_ids, add_time_ids, 1 + add_time_ids = train_util.concat_embeddings( + add_time_ids, add_time_ids, int(latents.shape[0]) ) else: # concat to fit batch size