Daemontatox commited on
Commit
4b69a7c
·
verified ·
1 Parent(s): 0435944

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -17,9 +17,9 @@ logging.basicConfig(level=logging.INFO)
17
  logger = logging.getLogger(__name__)
18
 
19
  # Load model and processor
20
- ckpt = "OpenGVLab/InternVL2_5-38B"
21
- model = AutoModel.from_pretrained(ckpt, torch_dtype=torch.bfloat16).to("cuda")
22
- processor = AutoProcessor.from_pretrained(ckpt)
23
 
24
  class DocumentState:
25
  def __init__(self):
 
17
  logger = logging.getLogger(__name__)
18
 
19
  # Load model and processor
20
+ ckpt = "OpenGVLab/InternVL2_5-38B-MPO"
21
+ model = AutoModel.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
22
+ processor = AutoProcessor.from_pretrained(ckpt,trust_remote_code=True)
23
 
24
  class DocumentState:
25
  def __init__(self):