Skip to content

Commit

Permalink
Increase timeout for lm-format-enforcer
Browse files Browse the repository at this point in the history
  • Loading branch information
rlouf committed Oct 16, 2024
1 parent caeaaf4 commit e7c5dc2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/benchmark_lfe.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit e7c5dc2

Please sign in to comment.