Skip to content

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

@Autumn5

Description

@Autumn5

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions