update txt/img/control size controls

This commit is contained in:
Vladimir Mandic
2024-09-08 18:49:49 -04:00
parent 709429be75
commit 62d3cdf37d
9 changed files with 40 additions and 17 deletions
+2 -1
View File
@@ -81,7 +81,8 @@ def diffusers_callback(pipe, step: int, timestep: int, kwargs: dict):
else:
pipe._guidance_scale = 0.0 # pylint: disable=protected-access
for key in {"prompt_embeds", "negative_prompt_embeds", "add_text_embeds", "add_time_ids"} & set(kwargs):
kwargs[key] = kwargs[key].chunk(2)[-1]
if kwargs[key] is not None:
kwargs[key] = kwargs[key].chunk(2)[-1]
try:
if hasattr(pipe, "_unpack_latents") and hasattr(pipe, "vae_scale_factor"): # FLUX
if p.hr_resize_mode > 0 and (p.hr_upscaler != 'None' or p.hr_resize_mode == 5):