Defaults for llm features

ProbablyFaiz · Sep 6, 2024 · 7cee793 · 7cee793
1 parent 6e45d05
commit 7cee793
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rl/llm/engines/core.py b/rl/llm/engines/core.py
@@ -377,11 +377,13 @@ def decorator(func):
         @click.option(
             "--json-output",
             is_flag=True,
+            default=defaults.get("json_output", False),
             help="Whether to output the response in JSON format.",
         )
         @click.option(
             "--return-logprobs",
             is_flag=True,
+            default=defaults.get("return_logprobs", False),
             help="Whether to return logprobs in the response.",
         )
         def wrapper(*args, **kwargs):