Update my_model/captioner/image_captioning.py
Browse files
my_model/captioner/image_captioning.py
CHANGED
|
@@ -36,7 +36,7 @@ class ImageCaptioningModel:
|
|
| 36 |
if self.model_type == 'i_blip':
|
| 37 |
self.processor = InstructBlipProcessor.from_pretrained(self.model_path,
|
| 38 |
load_in_8bit=self.load_in_8bit,
|
| 39 |
-
load_in_4bit=self.load_in_4bit
|
| 40 |
torch_dtype=self.torch_dtype,
|
| 41 |
device_map=self.device_map
|
| 42 |
)
|
|
|
|
| 36 |
if self.model_type == 'i_blip':
|
| 37 |
self.processor = InstructBlipProcessor.from_pretrained(self.model_path,
|
| 38 |
load_in_8bit=self.load_in_8bit,
|
| 39 |
+
load_in_4bit=self.load_in_4bit,
|
| 40 |
torch_dtype=self.torch_dtype,
|
| 41 |
device_map=self.device_map
|
| 42 |
)
|