yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-22 14:43:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

merge (#4620)

Browse Source
This commit is contained in:
ZanSara 2023-04-12 09:38:04 +02:00 committed by GitHub
parent 3d79174eb8
commit 1ac9ca7fac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
haystack/nodes/prompt/invocation_layer/chatgpt.py
View File

@ -111,5 +111,4 @@ class ChatGPTInvocationLayer(OpenAIInvocationLayer):
@classmethod
def supports(cls, model_name_or_path: str, **kwargs) -> bool:
valid_model = any(m for m in ["gpt-3.5-turbo"] if m in model_name_or_path)
return valid_model
return model_name_or_path in ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k"]

2
haystack/nodes/prompt/prompt_model.py
View File

@ -11,7 +11,7 @@ logger = logging.getLogger(__name__)
def instruction_following_models() -> List[str]:
return ["flan", "mt0", "bloomz", "davinci", "opt-iml"]
return ["flan", "mt0", "bloomz", "davinci", "opt-iml", "gpt-3.5-turbo", "gpt-4"]
class PromptModel(BaseComponent):