jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Add a way to set patches that modify the attn2 output.

Browse Source 
Change the transformer patches function format to be more future proof.
This commit is contained in:
comfyanonymous
2023-06-18 22:58:22 -04:00
parent cd930d4e7f
commit 8883cb0f67
4 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/sd.py
View File

@@ -331,6 +331,9 @@ class ModelPatcher:
def set_model_attn2_patch(self, patch):
self.set_model_patch(patch, "attn2_patch")
def set_model_attn2_output_patch(self, patch):
self.set_model_patch(patch, "attn2_output_patch")
def model_patches_to(self, device):
to = self.model_options["transformer_options"]
if "patches" in to: