bug fix

amazon-science · Sep 4, 2024 · d5cd604 · d5cd604
1 parent 566ab22
commit d5cd604
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/refchecker/utils.py b/refchecker/utils.py
@@ -100,7 +100,7 @@ def get_model_batch_response(
 
     if sagemaker_client is not None:
         parameters = {
-            "max_length": max_new_tokens,
+            "max_new_tokens": max_new_tokens,
             "temperature": temperature,
             "num_beams": 1,
             "do_sample": False,