Skip to content

GitLab

Commit 887a6341 authored by comfyanonymous's avatar comfyanonymous
Browse files

Proper ModelMergeSD3_2B node.

parent 1e2839f4
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
comfy_extras/nodes_model_merging_model_specific.py
View file @ 887a6341
...@@ -52,7 +52,7 @@ class ModelMergeSDXL(comfy_extras.nodes_model_merging.ModelMergeBlocks): ...@@ -52,7 +52,7 @@ class ModelMergeSDXL(comfy_extras.nodes_model_merging.ModelMergeBlocks):
return {"required": arg_dict} return {"required": arg_dict}
class ModelMergeSD3(comfy_extras.nodes_model_merging.ModelMergeBlocks): class ModelMergeSD3_2B(comfy_extras.nodes_model_merging.ModelMergeBlocks):
CATEGORY = "advanced/model_merging/model_specific" CATEGORY = "advanced/model_merging/model_specific"
@classmethod @classmethod
...@@ -68,7 +68,7 @@ class ModelMergeSD3(comfy_extras.nodes_model_merging.ModelMergeBlocks): ...@@ -68,7 +68,7 @@ class ModelMergeSD3(comfy_extras.nodes_model_merging.ModelMergeBlocks):
arg_dict["y_embedder."] = argument arg_dict["y_embedder."] = argument
arg_dict["t_embedder."] = argument arg_dict["t_embedder."] = argument
for i in range(38): for i in range(24):
arg_dict["joint_blocks.{}.".format(i)] = argument arg_dict["joint_blocks.{}.".format(i)] = argument
arg_dict["final_layer."] = argument arg_dict["final_layer."] = argument
...@@ -79,5 +79,5 @@ NODE_CLASS_MAPPINGS = { ...@@ -79,5 +79,5 @@ NODE_CLASS_MAPPINGS = {
"ModelMergeSD1": ModelMergeSD1, "ModelMergeSD1": ModelMergeSD1,
"ModelMergeSD2": ModelMergeSD1, #SD1 and SD2 have the same blocks "ModelMergeSD2": ModelMergeSD1, #SD1 and SD2 have the same blocks
"ModelMergeSDXL": ModelMergeSDXL, "ModelMergeSDXL": ModelMergeSDXL,
"ModelMergeSD3": ModelMergeSD3, "ModelMergeSD3_2B": ModelMergeSD3_2B,
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment