y0un92 commited on
Commit
2cfde2e
1 Parent(s): a34f995

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 = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5-int4', trust_remote_code=True)
29
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
30
  model.eval()
31
 
 
25
  assert device in ['cuda', 'mps']
26
 
27
  # Load model
28
+ model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5-int4', trust_remote_code=True, load_in_4bit=True)
29
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
30
  model.eval()
31