From 3c194764878a6c8fd0afae83c781ff47528d4005 Mon Sep 17 00:00:00 2001 From: Alessandro Sordoni Date: Wed, 14 Aug 2024 10:46:55 -0700 Subject: [PATCH] remove compute_transfer_matrix --- projects/modular_llm/train_experts.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/projects/modular_llm/train_experts.py b/projects/modular_llm/train_experts.py index c7bbaa9f9..43098cf27 100644 --- 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)