Merge pull request #500 from RockChinQ/perf-more-model-support

Perf more model support
This commit is contained in:
Junyan Qin
2023-06-16 19:40:29 +08:00
committed by GitHub
2 changed files with 11 additions and 6 deletions
+5 -3
View File
@@ -204,11 +204,13 @@ auto_reset = True
# 现已支持的模型有: # 现已支持的模型有:
# #
# 'gpt-4' # 'gpt-4'
# 'gpt-4-0314' # 'gpt-4-0613'
# 'gpt-4-32k' # 'gpt-4-32k'
# 'gpt-4-32k-0314' # 'gpt-4-32k-0613'
# 'gpt-3.5-turbo' # 'gpt-3.5-turbo'
# 'gpt-3.5-turbo-0301' # 'gpt-3.5-turbo-16k'
# 'gpt-3.5-turbo-0613'
# 'gpt-3.5-turbo-16k-0613'
# 'text-davinci-003' # 'text-davinci-003'
# 'text-davinci-002' # 'text-davinci-002'
# 'code-davinci-002' # 'code-davinci-002'
+6 -3
View File
@@ -20,11 +20,14 @@ COMPLETION_MODELS = {
CHAT_COMPLETION_MODELS = { CHAT_COMPLETION_MODELS = {
'gpt-3.5-turbo', 'gpt-3.5-turbo',
'gpt-3.5-turbo-0301', 'gpt-3.5-turbo-16k',
'gpt-3.5-turbo-0613',
'gpt-3.5-turbo-16k-0613',
# 'gpt-3.5-turbo-0301',
'gpt-4', 'gpt-4',
'gpt-4-0314', 'gpt-4-0613',
'gpt-4-32k', 'gpt-4-32k',
'gpt-4-32k-0314' 'gpt-4-32k-0613'
} }
EDIT_MODELS = { EDIT_MODELS = {