Эх сурвалжийг харах

Expand sampling in generate.py for new test

Eric Wang 3 жил өмнө
parent
commit
c3d94707ec
1 өөрчлөгдсөн 1 нэмэгдсэн , 1 устгасан
  1. 1 1
      generate.py

+ 1 - 1
generate.py

@@ -53,7 +53,7 @@ def evaluate(instruction, input=None, **kwargs):
         generation_config=generation_config,
         return_dict_in_generate=True,
         output_scores=True,
-        max_new_tokens=256,
+        max_new_tokens=2048,
     )
     s = generation_output.sequences[0]
     output = tokenizer.decode(s)