Spaces:
Runtime error
Runtime error
DongfuJiang
commited on
Commit
•
031687c
1
Parent(s):
e1fb4c8
Update app_high_res.py
Browse files- app_high_res.py +2 -2
app_high_res.py
CHANGED
@@ -13,8 +13,8 @@ from models.conversation import conv_templates
|
|
13 |
from typing import List
|
14 |
|
15 |
|
16 |
-
processor = AutoProcessor.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-high-res-
|
17 |
-
model = Idefics2ForConditionalGeneration.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-high-res-
|
18 |
MAX_NUM_FRAMES = 24
|
19 |
conv_template = conv_templates["idefics_2"]
|
20 |
|
|
|
13 |
from typing import List
|
14 |
|
15 |
|
16 |
+
processor = AutoProcessor.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-high-res-40k-mantis-2epoch_4096")
|
17 |
+
model = Idefics2ForConditionalGeneration.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-high-res-40k-mantis-2epoch_4096", torch_dtype=torch.bfloat16)
|
18 |
MAX_NUM_FRAMES = 24
|
19 |
conv_template = conv_templates["idefics_2"]
|
20 |
|