KB-VQA

Running

m7mdal7aj commited on Feb 22

Commit

1089b06

•

1 Parent(s): 398c0e8

Update my_model/captioner/image_captioning.py

Files changed (1) hide show

my_model/captioner/image_captioning.py CHANGED Viewed

@@ -36,7 +36,7 @@ class ImageCaptioningModel:
         if self.model_type == 'i_blip':
             self.processor = InstructBlipProcessor.from_pretrained(self.model_path,
                                                                    load_in_8bit=self.load_in_8bit,
-                                                                   load_in_4bit=self.load_in_4bit
                                                                    torch_dtype=self.torch_dtype,
                                                                    device_map=self.device_map
                                                                    )

         if self.model_type == 'i_blip':
             self.processor = InstructBlipProcessor.from_pretrained(self.model_path,
                                                                    load_in_8bit=self.load_in_8bit,
+                                                                   load_in_4bit=self.load_in_4bit,
                                                                    torch_dtype=self.torch_dtype,
                                                                    device_map=self.device_map
                                                                    )