Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|
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 |
|