From 9ef65b2192b58ccb64b81a2e1a2442f388ed6be3 Mon Sep 17 00:00:00 2001
From: Debopam Chowdhury Param
 <116789128+DebopamParam@users.noreply.github.com>
Date: Thu, 7 Nov 2024 23:34:57 +0530
Subject: [PATCH] Added method save_pretrained() to RAGMultiModalModel

---
 byaldi/RAGModel.py | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/byaldi/RAGModel.py b/byaldi/RAGModel.py
index a05db45..65dcf4c 100644
--- a/byaldi/RAGModel.py
+++ b/byaldi/RAGModel.py
@@ -198,15 +198,15 @@ def save_pretrained(self, directory_path: str) -> None:
         PermissionError: If there are insufficient permissions to write to the directory.
         Exception: For any other unexpected errors during the save process.
     """
-    try:
-        # Attempt to save the model and processor
-        self.model.model.save_pretrained(directory_path)
-        self.model.processor.save_pretrained(directory_path)
-        
-    except FileNotFoundError as fnf_error:
-        raise FileNotFoundError(f"The specified directory '{directory_path}' does not exist.") from fnf_error
-    except PermissionError as perm_error:
-        raise PermissionError(f"Insufficient permissions to write to '{directory_path}'.") from perm_error
-    except Exception as e:
-        raise Exception(f"An unexpected error occurred while saving: {e}") from e
+        try:
+            # Attempt to save the model and processor
+            self.model.model.save_pretrained(directory_path)
+            self.model.processor.save_pretrained(directory_path)
+            
+        except FileNotFoundError as fnf_error:
+            raise FileNotFoundError(f"The specified directory '{directory_path}' does not exist.") from fnf_error
+        except PermissionError as perm_error:
+            raise PermissionError(f"Insufficient permissions to write to '{directory_path}'.") from perm_error
+        except Exception as e:
+            raise Exception(f"An unexpected error occurred while saving: {e}") from e