diff --git a/src/benchmark_lfe.py b/src/benchmark_lfe.py index 7379de8..22a62cf 100644 --- a/src/benchmark_lfe.py +++ b/src/benchmark_lfe.py @@ -11,7 +11,7 @@ class LMFormatEnforcerRegex: params = [models, regex_cases.keys()] param_names = ["model", "regex_name"] - timeout = 600 + timeout = 1200 def setup(self, model, _): """Set up the benchmark. @@ -44,7 +44,7 @@ def teardown(self, *args): class LMFormatEnforcerJsonSchema: params = [models, json_cases.keys()] param_names = ["model", "json_schema_name"] - timeout = 600 + timeout = 1200 def setup(self, model, _): """Set up the benchmark.