revert the code

microsoft · Nov 10, 2024 · 836323b · 836323b
1 parent e8ff2fa
commit 836323b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mttl/models/lightning/expert_module.py b/mttl/models/lightning/expert_module.py
@@ -149,7 +149,7 @@ def on_save_checkpoint(self, ckpt):
         ).expert_info.asdict()
 
     def as_expert(self):
-        return self.model.as_expert(training_config=self.training_config)
+        return self.model.as_expert()
 
 
 class SPLITExpertModule(ExpertModule):