CheckpointInfo -> Checkpoint Selector, to be a little less confusing. Fix TripleLoraLoader.
This commit is contained in:
+2
-2
@@ -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
@@ -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
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user