diff --git a/tests/inference/python_test_configs/generate_configs.py b/tests/inference/python_test_configs/generate_configs.py
index 0a745c7984..2720304d4f 100644
--- a/tests/inference/python_test_configs/generate_configs.py
+++ b/tests/inference/python_test_configs/generate_configs.py
@@ -34,6 +34,7 @@
     "full_precision": True,
     "prompt": "",
     "output_file": "",
+    "max_length": 128,
 }
 ssm_configs = {
     "ssms": [