ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues 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:
# 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