298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Proper ModelMergeSD3_2B node.

Browse Source
This commit is contained in:
comfyanonymous 2024-06-21 08:41:31 -04:00
parent 1e2839f4d9
commit 887a6341ed
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_model_merging_model_specific.py
View File

@ -52,7 +52,7 @@ class ModelMergeSDXL(comfy_extras.nodes_model_merging.ModelMergeBlocks):
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"
@classmethod
@ -68,7 +68,7 @@ class ModelMergeSD3(comfy_extras.nodes_model_merging.ModelMergeBlocks):
arg_dict["y_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["final_layer."] = argument
@ -79,5 +79,5 @@ NODE_CLASS_MAPPINGS = {
"ModelMergeSD1": ModelMergeSD1,
"ModelMergeSD2": ModelMergeSD1, #SD1 and SD2 have the same blocks
"ModelMergeSDXL": ModelMergeSDXL,
"ModelMergeSD3": ModelMergeSD3,
"ModelMergeSD3_2B": ModelMergeSD3_2B,
}