github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed an issue with CFG time embeds on SDXL

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-15 18:02:14 -06:00
parent 55a5fcc7d9
commit 8d09eb44ec
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/stable_diffusion_model.py
View File

@@ -444,8 +444,8 @@ class StableDiffusion:
if do_classifier_free_guidance: if do_classifier_free_guidance:
# todo check this with larget batches # todo check this with larget batches
train_util.concat_embeddings( add_time_ids = train_util.concat_embeddings(
add_time_ids, add_time_ids, 1 add_time_ids, add_time_ids, int(latents.shape[0])
) )
else: else:
# concat to fit batch size # concat to fit batch size