y0un92 commited on
Commit
23ab17f
·
verified ·
1 Parent(s): e9d26b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ device = args.device
25
  assert device in ['cuda', 'mps']
26
 
27
  # Load model
28
- model_path = 'https://huggingface.co/openbmb/MiniCPM-Llama3-V-2_5'
29
  if 'int4' in model_path:
30
  if device == 'mps':
31
  print('Error: running int4 model with bitsandbytes on Mac is not supported right now.')
 
25
  assert device in ['cuda', 'mps']
26
 
27
  # Load model
28
+ model_path = AutoModel.from_pretrained("openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True)
29
  if 'int4' in model_path:
30
  if device == 'mps':
31
  print('Error: running int4 model with bitsandbytes on Mac is not supported right now.')