Add nearswap to registry

arcee-ai · Jan 25, 2025 · 976361b · 976361b
1 parent c1a81f7
commit 976361b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/mergekit/merge_methods/nearswap.py b/mergekit/merge_methods/nearswap.py
@@ -71,6 +71,15 @@ def execute(self, tensors: Dict[ModelReference, torch.Tensor]) -> torch.Tensor:
 
 
 class NearSwapMerge(MergeMethod):
+    def name(self) -> str:
+        return "nearswap"
+
+    def pretty_name(self) -> Optional[str]:
+        return "NearSwap"
+
+    def reference_url(self) -> Optional[str]:
+        return "https://huggingface.co/alchemonaut/QuartetAnemoi-70B-t0.0001"
+
     def parameters(self) -> List[ConfigParameterDef]:
         return [ConfigParameterDef(name="t", required=True)]
 

diff --git a/mergekit/merge_methods/registry.py b/mergekit/merge_methods/registry.py
@@ -22,6 +22,7 @@
 )
 from mergekit.merge_methods.linear import LinearMerge
 from mergekit.merge_methods.model_stock import ModelStockMerge
+from mergekit.merge_methods.nearswap import NearSwapMerge
 from mergekit.merge_methods.nuslerp import NuSlerpMerge
 from mergekit.merge_methods.passthrough import PassthroughMerge
 from mergekit.merge_methods.sce import SCEMerge
@@ -35,6 +36,7 @@
     PassthroughMerge(),
     ModelStockMerge(),
     SCEMerge(),
+    NearSwapMerge(),
     # generalized task arithmetic methods
     GeneralizedTaskArithmeticMerge(
         consensus_method=None,