Merge pull request #15 from CodingQinghao/main

fix output BUG
This commit is contained in:
Qian 2022-08-03 23:23:04 +08:00 коммит произвёл GitHub
Родитель 0cb0758570 cb427e9ddd
Коммит 8982c96659
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -36,4 +36,4 @@ class TAPEXModelInterface:
beam=5
)
# the result should be a list of answers, and we only care about the answer itself instead of score
return model_output[0][0]
return model_output