search

OpenBMB / VisCPM

[ICLR'24 spotlight] Chinese and English Multimodal Large Model Series (Chat and Paint) | 基于CPM基础模型的中英双语多模态大模型系列
1.06k stars 93 forks source link

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

Open Autumn5 opened 5 months ago

Autumn5 commented 5 months ago

求助： 我这样调用的， 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左右时就会被自动杀死，显卡够用，这是怎么回事呢？

JamesHujy commented 4 months ago

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