mirror of
https://github.com/RYDE-WORK/MiniCPM.git
synced 2026-01-19 21:03:39 +08:00
commit
d2ee9b6e0c
@ -137,7 +137,7 @@ import torch
|
||||
from PIL import Image
|
||||
from transformers import AutoModel, AutoTokenizer
|
||||
|
||||
model = AutoModel.from_pretrained('openbmb/MiniCPM-V/', trust_remote_code=True)
|
||||
model = AutoModel.from_pretrained('openbmb/MiniCPM-V', trust_remote_code=True)
|
||||
tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-V', trust_remote_code=True)
|
||||
model.eval().cuda()
|
||||
|
||||
|
||||
@ -143,7 +143,7 @@ import torch
|
||||
from PIL import Image
|
||||
from transformers import AutoModel, AutoTokenizer
|
||||
|
||||
model = AutoModel.from_pretrained('openbmb/MiniCPM-V/', trust_remote_code=True)
|
||||
model = AutoModel.from_pretrained('openbmb/MiniCPM-V', trust_remote_code=True)
|
||||
tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-V', trust_remote_code=True)
|
||||
model.eval().cuda()
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user