From 1cfd00f790f3d19ce4969d07f5c43ead6603022a Mon Sep 17 00:00:00 2001 From: Anke Tang Date: Mon, 13 Jan 2025 22:32:48 +0800 Subject: [PATCH] Update slerp.py --- fusion_bench/method/slerp/slerp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fusion_bench/method/slerp/slerp.py b/fusion_bench/method/slerp/slerp.py index d69ad2dd..321a37c3 100644 --- a/fusion_bench/method/slerp/slerp.py +++ b/fusion_bench/method/slerp/slerp.py @@ -51,7 +51,7 @@ class SlerpMergeAlgorithm(BaseAlgorithm): General purpose implementation of Slerp (Spherical Linear Interpolation) for PyTorch models. """ - _config_mapping = BaseAlgorithm._config_mapping + { + _config_mapping = BaseAlgorithm._config_mapping | { "t": "t", "DOT_THRESHOLD": "DOT_THRESHOLD", "epsilon": "epsilon",