大模型應(yīng)用:相似句子或文章生成
可在線運(yùn)行的notebook示例:再kaggle網(wǎng)站搜索 ExampleSentence
這里將采用ChatGLM2-6B-int4作為基底模型,需要安裝以下依賴包:
接著定義回復(fù)的類,用于初始化模型和生成相似句子:
定義完類之后,我們即可實(shí)例化類:(同時(shí)你還可以更改為huggingface中的其他模型)
現(xiàn)在,我們可以定義需要生成的數(shù)據(jù)格式:
最后,需要調(diào)用生成的方法,得到返回的列表:
標(biāo)簽: