cyberfish1120 / tds Goto Github PK

你好，程序在运行的时候出现以下错误
Traceback (most recent call last):
File "main.py", line 3, in
from ner_code.predict import NerPredict
File "/home/aistudio/PaddleTDS-master/ner_code/predict.py", line 5, in
from ner_code.ner_utils import model, utils_fn
File "/home/aistudio/PaddleTDS-master/ner_code/ner_utils/model.py", line 5, in
from paddlenlp.transformers import NeZhaModel
ImportError: cannot import name 'NeZhaModel' from 'paddlenlp.transformers' (/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddlenlp/transformers/init.py)
请问是还需要安装什么包吗？

询问价格时出现该错误，具体如下：

"用户： 你好，我要和同事一去北京办事，麻烦你帮我们找个可以行李寄存的酒店，我们希望酒店提供无烟房，并且评分是5分。
系统： 只找到一家叫如家商旅酒店(北京牡丹园地铁站店)。
用户： 那这家的酒店的价格是多少，我们公司有报销范围。"
Traceback (most recent call last):
File "main.py", line 24, in
agent.response(content)
File "/content/drive/.shortcut-targets-by-id/1xaW472DaLdA7OjwXQzUnm0qs-M9FEqo-/基于飞桨的中文多领域任务型对话系统/agent.py", line 34, in response
answer, self.DST = rule_response(content, intents, self.database, self.DST)
File "/content/drive/.shortcut-targets-by-id/1xaW472DaLdA7OjwXQzUnm0qs-M9FEqo-/基于飞桨的中文多领域任务型对话系统/response_code/response_main.py", line 16, in rule_response
responds.add(database.database_return(content, intent, DST))
File "/content/drive/.shortcut-targets-by-id/1xaW472DaLdA7OjwXQzUnm0qs-M9FEqo-/基于飞桨的中文多领域任务型对话系统/database_code/database_main.py", line 223, in database_return
if len(DST[intent]) > self.candidate_n:
TypeError: object of type 'int' has no len()

Out of memory error on GPU 0. Cannot allocate 28.125244MB memory on GPU 0, 10.749878GB memory has been allocated and available memory is only 11.562500MB.

Please check whether there is any other process using GPU 0.

1. If yes, please stop them, or start PaddlePaddle on another GPU.
2. If no, please decrease the batch size of your model.
   模型占用了10G多的显存，单块gpu无法运行，请问有其他的解决思路可参考么？