Skip to content

Commit 11ec7cf

Browse files
authored
default parameters for ollama_chat models (#689)
Signed-off-by: Mandana Vaziri <[email protected]>
1 parent 3c85bcd commit 11ec7cf

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

examples/chatbot/chatbot.pdl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ text:
66
- repeat:
77
text:
88
# Send context to Granite model hosted at replicate.com
9-
- model: ollama/granite-code:8b
9+
- model: ollama_chat/granite3.2:8b
1010
# Allow the user to type 'yes', 'no', or anything else, storing
1111
# the input into a variable named `eval`. The input is also implicitly
1212
# added to the context.

src/pdl/pdl_ast.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -902,6 +902,11 @@ def get_default_model_parameters() -> list[dict[str, Any]]:
902902
"temperature": 0,
903903
},
904904
},
905+
{
906+
"ollama_chat/granite*": {
907+
"temperature": 0,
908+
},
909+
},
905910
]
906911

907912

0 commit comments

Comments
 (0)