diff --git a/evals/evaluation/bigcode_evaluation_harness/arguments.py b/evals/evaluation/bigcode_evaluation_harness/arguments.py index b2303a47..8683873a 100644 --- a/evals/evaluation/bigcode_evaluation_harness/arguments.py +++ b/evals/evaluation/bigcode_evaluation_harness/arguments.py @@ -258,6 +258,7 @@ def __init__( check_references=False, user_model=None, # used for pass model object tokenizer=None, # used for pass tokenizer object + codegen_url=None, ): self.prefix = prefix self.do_sample = do_sample @@ -300,3 +301,4 @@ def __init__( self.check_references = check_references self.user_model = user_model self.tokenizer = tokenizer + self.codegen_url = codegen_url