Update __init__.py

arcee-ai · May 7, 2024 · fa0ed43 · fa0ed43
1 parent cbbd2cc
commit fa0ed43
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/mergekit/merge_methods/__init__.py b/mergekit/merge_methods/__init__.py
@@ -89,6 +89,20 @@ def get(method: str) -> MergeMethod:
             default_rescale=True,
             default_swapping=True,
         )
+    elif method == "breadcrumbs":
+        return GeneralizedTaskArithmeticMerge(
+            consensus_method=None,
+            sparsification_method=SparsificationMethod.magnitude_outliers,
+            default_normalize=False,
+            default_rescale=False,
+        )
+    elif method == "breadcrumbs_ties":
+        return GeneralizedTaskArithmeticMerge(
+            consensus_method=ConsensusMethod.sum,
+            sparsification_method=SparsificationMethod.magnitude_outliers,
+            default_normalize=False,
+            default_rescale=False,
+        )
     elif method == "model_stock":
         return ModelStockMerge()
     raise RuntimeError(f"Unimplemented merge method {method}")