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

Scheduling and cleanup

Browse Source
This commit is contained in:
AI-Casanova
2024-10-28 22:32:26 -05:00
parent f3442abc92
commit 39dfa9cbdb
3 changed files with 39 additions and 158 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_callbacks.py
+8 -8
View File
@@ -67,14 +67,14 @@ def diffusers_callback(pipe, step: int, timestep: int, kwargs: dict):
pipe.set_ip_adapter_scale(ip_adapter_scales)
if step != getattr(pipe, 'num_timesteps', 0):
kwargs = processing_correction.correction_callback(p, timestep, kwargs)
# if p.scheduled_prompt and 'prompt_embeds' in kwargs and 'negative_prompt_embeds' in kwargs:
# try:
# i = (step + 1) % len(p.prompt_embeds)
# kwargs["prompt_embeds"] = p.prompt_embeds[i][0:1].expand(kwargs["prompt_embeds"].shape)
# j = (step + 1) % len(p.negative_embeds)
# kwargs["negative_prompt_embeds"] = p.negative_embeds[j][0:1].expand(kwargs["negative_prompt_embeds"].shape)
# except Exception as e:
# shared.log.debug(f"Callback: {e}")
if p.embedder is not None:
try:
if 'prompt_embeds' in kwargs:
kwargs["prompt_embeds"] = p.embedder("prompt_embeds", step + 1)
if 'negative_prompt_embeds' in kwargs:
kwargs["negative_prompt_embeds"] = p.embedder("negative_prompt_embeds", step + 1)
except Exception as e:
shared.log.debug(f"Callback: {e}")
if step == int(getattr(pipe, 'num_timesteps', 100) * p.cfg_end) and 'prompt_embeds' in kwargs and 'negative_prompt_embeds' in kwargs:
if "PAG" in shared.sd_model.__class__.__name__:
pipe._guidance_scale = 1.001 if pipe._guidance_scale > 1 else pipe._guidance_scale # pylint: disable=protected-access