From cc9725fcd343d2f4ead8874b4018d75fe723fcab Mon Sep 17 00:00:00 2001
From: Martin Bukowski <mbukowski@bmds.us>
Date: Tue, 2 Jan 2024 11:48:17 -0600
Subject: [PATCH] housekeeping

---
 daremerge.py | 9 +++------
 magmerge.py  | 2 +-
 2 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/daremerge.py b/daremerge.py
index 5afe04b..be674ce 100644
--- a/daremerge.py
+++ b/daremerge.py
@@ -1,4 +1,4 @@
-# nodes.py
+# daremerge.py
 import torch
 from typing import Dict, Tuple, Optional
 
@@ -200,11 +200,8 @@ class DareModelMerger:
         return mask
 
     def apply_sparsification(self, base_model_param: torch.Tensor, model_a_param: torch.Tensor, model_b_param: torch.Tensor,
-                             exclude_a: float, include_b: float,
-                             invert : str, drop_rate: float,
-                             seed: Optional[int] = None,
-                             clear_cache : bool = False,
-                             **kwargs) -> torch.Tensor:
+                             exclude_a: float, include_b: float, invert : str, drop_rate: float,
+                             seed: Optional[int] = None, **kwargs) -> torch.Tensor:
         """
         Applies sparsification to a tensor based on the specified sparsity level.
         """
diff --git a/magmerge.py b/magmerge.py
index f1971b3..5893e25 100644
--- a/magmerge.py
+++ b/magmerge.py
@@ -1,4 +1,4 @@
-# nodes.py
+# magmerge.py
 import torch
 from typing import Dict, Tuple, Optional