VisCPM icon indicating copy to clipboard operation
VisCPM copied to clipboard
verified publisher icon OpenBMB

求助--加载本地模型代码出错

Open Autumn5 opened this issue 1 year ago • 1 comments

求助： 我这样调用的， from VisCPM import VisCPMChat from PIL import Image

model_path = './openbmb/VisCPM-Chat/pytorch_model.zhplus.bin' viscpm_chat = VisCPMChat(model_path, image_safety_checker=True)

我模型加载的路径对吗？ 加载模型中，GPU到811M左右时就会被自动杀死，显卡够用，这是怎么回事呢？

Autumn5 avatar Mar 11 '24 12:03 Autumn5

您好，您可以看一下是否是内存限制的原因

JamesHujy avatar Apr 22 '24 11:04 JamesHujy