from lmdeploy import pipeline, TurbomindEngineConfig
 backend_config = TurbomindEngineConfig(tp=8,model_format='hf', session_len=4096,cache_max_entry_count=0.2)


model_path="/data2//open_model/llava-v1.6-vicuna-7b"
#模型路径是指模型文件所在的本地目录路径,或者是在huggingface.co上托管的模型仓库中的模型ID。

pipe = pipeline(model_path,backend_config=backend_config)
image = load_image('https://raw.githubusercontent.com/open-mmlab/mmdeploy/main/tests/data/tiger.jpeg')
    print("image",image)
 response = pipe(('describe this image', image))
 print("response",response.text)

分析得出是输入长度加上图片token过程,vicuna 默认的session_len长度(2048)过短造成的,对话模版也需要了解。
在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐