Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

update changelog and cleanup

Browse Source
This commit is contained in:
Vladimir Mandic
2024-08-29 20:16:32 -04:00
parent 81583efdf5
commit 1114165cc1
4 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_callbacks.py
+1 -1
View File
@@ -83,7 +83,7 @@ def diffusers_callback(pipe, step: int, timestep: int, kwargs: dict):
for key in {"prompt_embeds", "negative_prompt_embeds", "add_text_embeds", "add_time_ids"} & set(kwargs):
kwargs[key] = kwargs[key].chunk(2)[-1]
if hasattr(pipe, "_unpack_latents") and hasattr(pipe, "vae_scale_factor"): # FLUX
shared.state.current_latent = pipe._unpack_latents(kwargs['latents'], p.height, p.width, pipe.vae_scale_factor)
shared.state.current_latent = pipe._unpack_latents(kwargs['latents'], p.height, p.width, pipe.vae_scale_factor) # pylint: disable=protected-access
else:
shared.state.current_latent = kwargs['latents']
if shared.cmd_opts.profile and shared.profiler is not None: