Add files via upload

2024-01-22 06:39:12 +09:00
parent bb921c2335
commit 0affb2093a
1 changed files with 4 additions and 0 deletions
@@ -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)",
 }