Open Autumn5 opened 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左右时就会被自动杀死,显卡够用,这是怎么回事呢?
您好,您可以看一下是否是内存限制的原因
求助: 我这样调用的, 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左右时就会被自动杀死,显卡够用,这是怎么回事呢?