Spaces:
Sleeping
Sleeping
NGUYEN, Xuan Phi
commited on
Commit
•
24d2b98
1
Parent(s):
df47d5c
update
Browse files
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')
|