rahul5035 commited on
Commit
fdcaf03
·
verified ·
1 Parent(s): 9ef5904

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import torch
9
  torch.cuda.empty_cache()
10
 
11
  # Load the model and tokenizer on CPU
12
- model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5-int4', trust_remote_code=True).to('cpu')
13
  tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5-int4', trust_remote_code=True)
14
  model.eval()
15
 
 
9
  torch.cuda.empty_cache()
10
 
11
  # Load the model and tokenizer on CPU
12
+ model = AutoModel.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5-int4', trust_remote_code=True)
13
  tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-Llama3-V-2_5-int4', trust_remote_code=True)
14
  model.eval()
15