fix: update chat_func fix int issue

[chinwh2019]

描述

描述您的pull request所做的更改。
另外请附上相关程序运行时的截图（before & after），以直观地展现您的更改达成的效果。

相关问题

（如有）请列出与此拉取请求相关的issue。
或者，您可以直接描述您的工作能解决的问题等。

补充信息

请提供任何其他信息或说明，有助于其他贡献者理解您的更改。

如果您提交的是 draft pull request，也请在这里写明开发进度。例子：

开发进度

• 已经做好的事情1
• 还要干的事情1
• 还要干的事情2

[chinwh2019]

Hello, I am submitting another pull request. I just cast the int <total_token_count> to list in chat_func, predict_all

[GaiZhenbiao]

好！

Edit：合并了。