search

WisdomShell / codeshell

A series of code large language models developed by PKU-KCL
http://se.pku.edu.cn/kcl
Other
1.61k stars 119 forks source link

想问下该模型支持pipeline加载吗？如果支持，task是什么类型，我用的chat提示不text-generation类型？ #65

Open lvqing001 opened 8 months ago

ruixie commented 6 months ago

抱歉，我们暂时不支持pipeline，你可以通过model.chat方法来与模型进行交互。 https://github.com/WisdomShell/codeshell/blob/3c96e3458a6a385e2473002e3fd5fb4870442ca7/model/modeling_codeshell.py#L954