mirror of
https://github.com/RYDE-WORK/MiniCPM.git
synced 2026-02-01 12:25:38 +08:00
Update README.md
Fix: model path error for MiniCPM-V.
This commit is contained in:
parent
600e00dba3
commit
6b14a7e50d
@ -143,7 +143,7 @@ import torch
|
|||||||
from PIL import Image
|
from PIL import Image
|
||||||
from transformers import AutoModel, AutoTokenizer
|
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)
|
tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-V', trust_remote_code=True)
|
||||||
model.eval().cuda()
|
model.eval().cuda()
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user