CheckpointInfo -> Checkpoint Selector, to be a little less confusing. Fix TripleLoraLoader.

This commit is contained in:
arcum42
2025-03-29 18:12:45 -07:00
parent 5c648f3b7b
commit e5ccb278b8
7 changed files with 8 additions and 8 deletions
+2 -2
View File
@@ -30,6 +30,7 @@ NODE_CLASS_MAPPINGS = {
"Sage_PonyPrefix": Sage_PonyPrefix,
# Model nodes
"Sage_CheckpointSelector": Sage_CheckpointSelector, #
"Sage_UNETLoader": Sage_UNETLoader,
"Sage_CheckpointLoaderSimple": Sage_CheckpointLoaderSimple,
"Sage_CheckpointLoaderRecent": Sage_CheckpointLoaderRecent,
@@ -72,7 +73,6 @@ NODE_CLASS_MAPPINGS = {
# Utility nodes
"Sage_GetFileHash": Sage_GetFileHash,
"Sage_CheckpointInfoOnly": Sage_CheckpointInfoOnly, #
# Test nodes
"Sage_Foobar": Sage_Foobar#,
@@ -101,6 +101,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"Sage_PonyPrefix": "Add Pony v6 Prefixes",
# Model nodes
"Sage_CheckpointSelector": "Checkpoint Selector", #
"Sage_UNETLoader": "Load Diffusion Model w/ Metadata",
"Sage_CheckpointLoaderSimple": "Load Checkpoint w/ Metadata",
"Sage_CheckpointLoaderRecent": "Load Recently Used Checkpoint",
@@ -143,7 +144,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
# Utility nodes
"Sage_GetFileHash": "Get Sha256 Hash",
"Sage_CheckpointInfoOnly": "Checkpoint Info", #
# Test nodes
"Sage_Foobar": "Test Node (Do not use)"#,
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
+1 -1
View File
@@ -105,7 +105,7 @@ class Sage_TripleLoraStack(ComfyNodeABC):
def add_lora_to_stack(self, **args) -> tuple:
stack = args.get("lora_stack", None)
for i in range(1, len(args) // 4 + 1):
for i in range(1, len(args) // 5 + 1): # including refresh button
enabled = args[f"enabled_{i}"]
lora_name = args[f"lora_{i}_name"]
model_weight = args[f"model_{i}_weight"]
+1 -1
View File
@@ -100,7 +100,7 @@ class Sage_UNETLoader(UNETLoader):
model_info["hash"] = cache.data[model_info["path"]]["hash"]
return (loaders.unet(model_info["path"], weight_dtype), model_info)
class Sage_CheckpointInfoOnly(ComfyNodeABC):
class Sage_CheckpointSelector(ComfyNodeABC):
@classmethod
def INPUT_TYPES(cls) -> InputTypeDict:
return {