comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-22 23:49:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Introduced Resources to ComfyNodeV3

Browse Source
This commit is contained in:
Jedrzej Kosinski
2025-06-28 15:47:02 -07:00
parent 2999212480
commit 0e7ff98e1d
3 changed files with 62 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
comfy_extras/nodes_v3_test.py
View File

@@ -100,10 +100,6 @@ class V3TestNode(io.ComfyNodeV3):
class V3LoraLoader(io.ComfyNodeV3):
class State(io.NodeState):
loaded_lora: tuple[str, Any] | None = None
state: State
@classmethod
def DEFINE_SCHEMA(cls):
return io.SchemaV3(
@@ -147,17 +143,7 @@ class V3LoraLoader(io.ComfyNodeV3):
if strength_model == 0 and strength_clip == 0:
return io.NodeOutput(model, clip)
lora_path = folder_paths.get_full_path_or_raise("loras", lora_name)
lora = None
if cls.state.loaded_lora is not None:
if cls.state.loaded_lora[0] == lora_path:
lora = cls.state.loaded_lora[1]
else:
cls.state.loaded_lora = None
if lora is None:
lora = comfy.utils.load_torch_file(lora_path, safe_load=True)
cls.state.loaded_lora = (lora_path, lora)
lora = cls.resources.get_torch_dict(io.ResourceKeyFolderFilename("loras", lora_name))
model_lora, clip_lora = comfy.sd.load_lora_for_models(model, clip, lora, strength_model, strength_clip)
return io.NodeOutput(model_lora, clip_lora)