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