Update my_model/captioner/image_captioning.py

my_model/captioner/image_captioning.py

@@ -29,6 +29,10 @@ class ImageCaptioningModel:
 
 
     def load_model(self):
+
+        if self.load_in_4bit and self.load_in_8bit:  # check if in case both set to True by mistake.
+            self.load_in_4bit = False
+            
         if self.model_type == 'i_blip':
             self.processor = InstructBlipProcessor.from_pretrained(self.model_path,
                                                                    load_in_8bit=self.load_in_8bit,