xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-30 15:47:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added get_attachment func on ModelPatcher

Browse Source
This commit is contained in:
kosinkadink1@gmail.com 2024-09-24 16:20:53 +09:00
parent da6c0455cc
commit c422553b0b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_patcher.py
View File

@ -823,10 +823,13 @@ class ModelPatcher:
def set_attachments(self, key: str, attachment): def set_attachments(self, key: str, attachment):
self.attachments[key] = attachment self.attachments[key] = attachment
def remove_attachments(self, key: str): def remove_attachments(self, key: str):
if key in self.attachments: if key in self.attachments:
self.attachments.pop(key) self.attachments.pop(key)
def get_attachment(self, key: str):
return self.attachments.get(key, None)
def set_injections(self, key: str, injections: List[PatcherInjection]): def set_injections(self, key: str, injections: List[PatcherInjection]):
self.injections[key] = injections self.injections[key] = injections