From fa0ed433f500bedf84217c7247891e6639765a65 Mon Sep 17 00:00:00 2001 From: Ar57m <132871733+Ar57m@users.noreply.github.com> Date: Tue, 7 May 2024 15:57:54 -0300 Subject: [PATCH] Update __init__.py --- mergekit/merge_methods/__init__.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/mergekit/merge_methods/__init__.py b/mergekit/merge_methods/__init__.py index 883103aa..9b06c12d 100644 --- 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}")