hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-09 19:06:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(model_providers/ollama): Fix OllamaLargeLanguageModel to correctly set the stop option (#5217)

Browse Source
This commit is contained in:
-LAN-
2024-06-14 18:26:14 +08:00
committed by GitHub
parent b7ff765d8d
commit 7f44e88eda
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_runtime/model_providers/ollama/llm/llm.py
View File

@@ -195,7 +195,7 @@ class OllamaLargeLanguageModel(LargeLanguageModel):
data["options"] = model_parameters or {} data["options"] = model_parameters or {}
if stop: if stop:
data["stop"] = "\n".join(stop) data["options"]["stop"] = stop
completion_type = LLMMode.value_of(credentials["mode"]) completion_type = LLMMode.value_of(credentials["mode"])