From 0affb2093a1f5fdbbc06924a5d595fbba319ee5e Mon Sep 17 00:00:00 2001 From: a-ru2016 <124528449+a-ru2016@users.noreply.github.com> Date: Mon, 22 Jan 2024 06:39:12 +0900 Subject: [PATCH] Add files via upload --- __init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/__init__.py b/__init__.py index 19c95c7..0e4d63f 100644 --- a/__init__.py +++ b/__init__.py @@ -1,15 +1,19 @@ from .merge.dare import DareModelMerger +from .merge.dareMBW import DareModelMergerMBW from .merge.mag import MagnitudePruningModelMerger from .merge.block import BlockModelMergerAdv + NODE_CLASS_MAPPINGS = { "DareModelMerger": DareModelMerger, + "DareModelMergerMBW": DareModelMergerMBW, "MagnitudeModelMerger": MagnitudePruningModelMerger, "BlockModelMergerAdv": BlockModelMergerAdv, } NODE_DISPLAY_NAME_MAPPINGS = { "DareModelMerger": "ModelMergeByDARE", + "DareModelMergerMBW": "ModelMergeByDAREMBW", "MagnitudeModelMerger": "ModelMergeByMagnitudePruning", "BlockModelMergerAdv": "ModelMergeByBlock (Advanced)", }