Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -24,11 +24,11 @@ class KBVQA():
 
  
     def create_bnb_config(self) -> BitsAndBytesConfig:
-    """
-    Creates a BitsAndBytes configuration based on the quantization setting.
-    Returns:
-        BitsAndBytesConfig: Configuration for BitsAndBytes optimized model.
-    """
+        """
+        Creates a BitsAndBytes configuration based on the quantization setting.
+        Returns:
+            BitsAndBytesConfig: Configuration for BitsAndBytes optimized model.
+        """
         if self.quantization == '4bit':
             return BitsAndBytesConfig(
                 load_in_4bit=True,