DongfuJiang commited on
Commit
e7742fc
2 Parent(s): 0a842a1 031687c

Merge branch 'main' of https://huggingface.co/spaces/Mantis-VL/Mantis-Video-eval

Browse files
Files changed (1) hide show
  1. 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-35k-mantis-2epoch_4096")
17
- model = Idefics2ForConditionalGeneration.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-high-res-35k-mantis-2epoch_4096", torch_dtype=torch.bfloat16)
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