报错信息为:
conv = conv_templates[args.conv_mode].copy()
KeyError: 'conv_template'
print中间过程的输出,发现加载模型时没获取到args.conv_mode,这个参数的获取代码文件在MoE-LLaVA/moellava/serve/cli.py。
于是考虑在报错点上面直接加上定义。
args.conv_mode = qwen
或者
conv = conv_templates['qwen'].copy()
这里我用的LLM是qwen,可以根据自己的代码选择具体参数,比如:openchat,phi,stablelm,llava_llama_2,llava_v1,mpt,llava_v0