NGUYEN, Xuan Phi commited on
Commit
24d2b98
1 Parent(s): df47d5c
multipurpose_chatbot/engines/sealava16_transformers_engine.py CHANGED
@@ -180,7 +180,7 @@ class SeaLlava16Engine(TransformersEngine):
180
  # /custom
181
 
182
  self._tokenizer = AutoTokenizer.from_pretrained(model_path)
183
- self._image_processor = LlavaNextImageProcessor(model_path)
184
  self._processor = LlavaNextProcessor(image_processor=self._image_processor, tokenizer=self._tokenizer)
185
  self._model = LlavaNextForConditionalGeneration.from_pretrained(model_path, torch_dtype=torch.bfloat16, device_map="cuda").eval()
186
  print(f'Loading llava1.6 from custom code')
 
180
  # /custom
181
 
182
  self._tokenizer = AutoTokenizer.from_pretrained(model_path)
183
+ self._image_processor = LlavaNextImageProcessor.from_pretrained(model_path)
184
  self._processor = LlavaNextProcessor(image_processor=self._image_processor, tokenizer=self._tokenizer)
185
  self._model = LlavaNextForConditionalGeneration.from_pretrained(model_path, torch_dtype=torch.bfloat16, device_map="cuda").eval()
186
  print(f'Loading llava1.6 from custom code')