Faustrix commited on
Commit
63a77c1
1 Parent(s): f11b25f
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModel, AutoTokenizer
5
  device = "cpu"
6
 
7
  # Load the model
8
- model = AutoModel.from_pretrained("openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True, torch_dtype=torch.float32)
9
  model = model.to(device=device)
10
 
11
  # Load the tokenizer
 
5
  device = "cpu"
6
 
7
  # Load the model
8
+ model = AutoModel.from_pretrained("openbmb/MiniCPM-Llama3-V-2_5", trust_remote_code=True)
9
  model = model.to(device=device)
10
 
11
  # Load the tokenizer