mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-15 05:57:20 +00:00
Model merging nodes for cosmos.
This commit is contained in:
parent
932ae8d9ca
commit
ed4d92b721
@ -196,6 +196,54 @@ class ModelMergeLTXV(comfy_extras.nodes_model_merging.ModelMergeBlocks):
|
|||||||
|
|
||||||
return {"required": arg_dict}
|
return {"required": arg_dict}
|
||||||
|
|
||||||
|
class ModelMergeCosmos7B(comfy_extras.nodes_model_merging.ModelMergeBlocks):
|
||||||
|
CATEGORY = "advanced/model_merging/model_specific"
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def INPUT_TYPES(s):
|
||||||
|
arg_dict = { "model1": ("MODEL",),
|
||||||
|
"model2": ("MODEL",)}
|
||||||
|
|
||||||
|
argument = ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01})
|
||||||
|
|
||||||
|
arg_dict["pos_embedder."] = argument
|
||||||
|
arg_dict["extra_pos_embedder."] = argument
|
||||||
|
arg_dict["x_embedder."] = argument
|
||||||
|
arg_dict["t_embedder."] = argument
|
||||||
|
arg_dict["affline_norm."] = argument
|
||||||
|
|
||||||
|
|
||||||
|
for i in range(28):
|
||||||
|
arg_dict["blocks.block{}.".format(i)] = argument
|
||||||
|
|
||||||
|
arg_dict["final_layer."] = argument
|
||||||
|
|
||||||
|
return {"required": arg_dict}
|
||||||
|
|
||||||
|
class ModelMergeCosmos14B(comfy_extras.nodes_model_merging.ModelMergeBlocks):
|
||||||
|
CATEGORY = "advanced/model_merging/model_specific"
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def INPUT_TYPES(s):
|
||||||
|
arg_dict = { "model1": ("MODEL",),
|
||||||
|
"model2": ("MODEL",)}
|
||||||
|
|
||||||
|
argument = ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01})
|
||||||
|
|
||||||
|
arg_dict["pos_embedder."] = argument
|
||||||
|
arg_dict["extra_pos_embedder."] = argument
|
||||||
|
arg_dict["x_embedder."] = argument
|
||||||
|
arg_dict["t_embedder."] = argument
|
||||||
|
arg_dict["affline_norm."] = argument
|
||||||
|
|
||||||
|
|
||||||
|
for i in range(36):
|
||||||
|
arg_dict["blocks.block{}.".format(i)] = argument
|
||||||
|
|
||||||
|
arg_dict["final_layer."] = argument
|
||||||
|
|
||||||
|
return {"required": arg_dict}
|
||||||
|
|
||||||
NODE_CLASS_MAPPINGS = {
|
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
|
||||||
@ -206,4 +254,6 @@ NODE_CLASS_MAPPINGS = {
|
|||||||
"ModelMergeSD35_Large": ModelMergeSD35_Large,
|
"ModelMergeSD35_Large": ModelMergeSD35_Large,
|
||||||
"ModelMergeMochiPreview": ModelMergeMochiPreview,
|
"ModelMergeMochiPreview": ModelMergeMochiPreview,
|
||||||
"ModelMergeLTXV": ModelMergeLTXV,
|
"ModelMergeLTXV": ModelMergeLTXV,
|
||||||
|
"ModelMergeCosmos7B": ModelMergeCosmos7B,
|
||||||
|
"ModelMergeCosmos14B": ModelMergeCosmos14B,
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user