DongfuJiang commited on
Commit
031687c
1 Parent(s): e1fb4c8

Update app_high_res.py

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