wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-15 07:52:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

need to put extra options on the model instead of patcher

Browse Source
This commit is contained in:
Raphael Walker 2023-12-06 14:51:04 +01:00
parent 2920eab33b
commit 3cb1e5dc71
No known key found for this signature in database
GPG Key ID: E6F58BE3395D3AA8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_sag.py
View File

@ -64,12 +64,12 @@ class SagNode:
def patch(self, model, scale, blur_sigma): def patch(self, model, scale, blur_sigma):
m = model.clone() m = model.clone()
# set extra options on the model # set extra options on the model
m.extra_options["sag"] = True m.model.extra_options["sag"] = True
m.extra_options["sag_scale"] = scale m.model.extra_options["sag_scale"] = scale
m.extra_options["sag_sigma"] = blur_sigma m.model.extra_options["sag_sigma"] = blur_sigma
attn_scores = None attn_scores = None
m.get_attn_scores = lambda: attn_scores m.model.get_attn_scores = lambda: attn_scores
def attn_and_record(q, k, v, extra_options): def attn_and_record(q, k, v, extra_options):
nonlocal attn_scores nonlocal attn_scores