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

Fix PuLID

Browse Source
This commit is contained in:
Disty0
2024-12-26 19:26:33 +03:00
parent 3359792762
commit 2693a46f67
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+2
View File
@@ -1349,6 +1349,8 @@ def set_diffusers_attention(pipe):
def add_noise_pred_to_diffusers_callback(pipe):
if not hasattr(pipe, "_callback_tensor_inputs"):
return pipe
if pipe.__class__.__name__.startswith("StableDiffusion"):
pipe._callback_tensor_inputs.append("noise_pred") # pylint: disable=protected-access
elif pipe.__class__.__name__.startswith("StableCascade"):