fix error when use farui-plus model (#7316)

Co-authored-by: 雪风 <xuefeng@shifaedu.cn>
This commit is contained in:
噢哎哟喂 2024-08-15 12:14:13 +00:00 committed by GitHub
parent d1a6702aa4
commit 6fdbc7dbf3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -159,6 +159,8 @@ You should also complete the text started with ``` but not tell ``` directly.
"""
if model in ['qwen-turbo-chat', 'qwen-plus-chat']:
model = model.replace('-chat', '')
if model == 'farui-plus':
model = 'qwen-farui-plus'
if model in self.tokenizers:
tokenizer = self.tokenizers[model]