Remove default from config_templates

mozilla-ai · Feb 6, 2025 · 9cb9aa3 · 9cb9aa3
1 parent 6a51183
commit 9cb9aa3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lumigator/backend/backend/config_templates.py b/lumigator/backend/backend/config_templates.py
@@ -99,7 +99,7 @@
         "use_fast": "{use_fast}",
         "trust_remote_code": "{trust_remote_code}",
         "torch_dtype": "{torch_dtype}",
-        "max_new_tokens": 500
+        "max_new_tokens": "{max_new_tokens}"
     }},
      "job": {{
         "max_samples": {max_samples},