New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

Autumn5 opened this issue Mar 11, 2024 · 1 comment

Autumn5 commented Mar 11, 2024 •

edited

Loading

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

Collaborator

JamesHujy commented Apr 22, 2024

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment