From d6aa0e24b067e0a50c7c30cc6993437aca3e358a Mon Sep 17 00:00:00 2001 From: shaheennabi Date: Tue, 3 Dec 2024 09:38:34 -0800 Subject: [PATCH] updated --- src/finetuning/merge_base_and_finetuned_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/finetuning/merge_base_and_finetuned_model.py b/src/finetuning/merge_base_and_finetuned_model.py index d3d5479..323d285 100644 --- 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")