Set default params for watsonx chat and text gen

Signed-off-by: Mandana Vaziri <[email protected]>
IBM · Jan 24, 2025 · 02ee1cb · 02ee1cb
1 parent fe5ccf7
commit 02ee1cb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/pdl/pdl_ast.py b/src/pdl/pdl_ast.py
@@ -581,7 +581,12 @@ def get_default_model_parameters() -> list[dict[str, Any]]:
     """Model-specific defaults to apply"""
     return [
         {
-            "*watsonx*": {
+            "*watsonx/*": {
+                "temperature": 0,
+            },
+        },
+        {
+            "*watsonx_text*": {
                 "decoding_method": DECODING_METHOD,
                 "max_tokens": MAX_NEW_TOKENS,
                 "min_new_tokens": MIN_NEW_TOKENS,