mirror of
https://github.com/RYDE-WORK/MiniCPM.git
synced 2026-01-31 11:53:13 +08:00
commit
d2ee9b6e0c
@ -137,7 +137,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()
|
||||||
|
|
||||||
|
|||||||
@ -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