remove compute_transfer_matrix

microsoft · Aug 14, 2024 · 3c19476 · 3c19476
1 parent 782d84a
commit 3c19476
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/projects/modular_llm/train_experts.py b/projects/modular_llm/train_experts.py
@@ -192,9 +192,6 @@ def upload_library(expert_library, module):
 
         upload_library(expert_library, module)
 
-        if args.create_transfer_matrix:
-            create_transfer_matrix(args, checkpoint)
-
 
 if __name__ == "__main__":
     train_experts(ExpertConfig.parse(), ExpertModel)