updated

shaheennabi · Dec 3, 2024 · d6aa0e2 · d6aa0e2
1 parent 781f7ee
commit d6aa0e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/finetuning/merge_base_and_finetuned_model.py b/src/finetuning/merge_base_and_finetuned_model.py
@@ -14,10 +14,10 @@ def __init__(self):
             self.model_trainer = ModelTrainer()
 
             # Load the quantized model and tokenizer from ModelLoader
-            self.quantized_model, self.tokenizer = self.model_loader.initiate_model_loader()
+            self.quantized_model, _ = self.model_loader.initiate_model_loader()
 
             # Load the finetuned model and tokenizer from ModelTrainer
-            self.finetuned_model, _ = self.model_trainer.initiate_model_trainer()  # No need to use tokenizer here
+            self.finetuned_model, self.tokenizer = self.model_trainer.initiate_model_trainer()  # No need to use tokenizer here
 
             logging.info("Models and Tokenizers Loaded Successfully")