xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-30 16:47:12 +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

3
comfy/model_patcher.py
View File

@ -828,6 +828,9 @@ class ModelPatcher:
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