From b3ef9dfe60562a3a271a7dcb8730c77dc86c4f51 Mon Sep 17 00:00:00 2001 From: Luke Meyers Date: Thu, 18 Jul 2024 21:46:02 -0400 Subject: [PATCH] Remove comments --- mergekit/scripts/ABM/extract_activations.py | 5 ----- mergekit/scripts/ABM/extract_permutation_matrices.py | 2 -- 2 files changed, 7 deletions(-) diff --git a/mergekit/scripts/ABM/extract_activations.py b/mergekit/scripts/ABM/extract_activations.py index 96b7695b..7cb5961b 100644 --- a/mergekit/scripts/ABM/extract_activations.py +++ b/mergekit/scripts/ABM/extract_activations.py @@ -132,11 +132,6 @@ def main( model_config = model.config() model_arch_info = get_architecture_info(model_config) - # things to do: find the residual space - # rest difference the ignore_spaces - # residual space is the one to for hidden states - # the rest we will attach hooks based on the module name - _json = model_arch_info.definition residual_space = None diff --git a/mergekit/scripts/ABM/extract_permutation_matrices.py b/mergekit/scripts/ABM/extract_permutation_matrices.py index 84a98e4b..75c58692 100644 --- a/mergekit/scripts/ABM/extract_permutation_matrices.py +++ b/mergekit/scripts/ABM/extract_permutation_matrices.py @@ -91,10 +91,8 @@ def match_tensors_permute_MHA( corr_submatrix, maximize=True ) - # store cost (cost is maximized here) costs[j, k] = corr_submatrix[row_ind, col_ind].sum() - # store perm so we don't have to recompute it later col_inds_storage[j][k] = col_ind outer_row_ind, outer_col_ind = scipy.optimize.linear_sum_assignment(