Spaces:

naot97
/

zalo_test_1

Sleeping

naot97 commited on Jul 16, 2023

Commit

9e21612

•

1 Parent(s): d4238f5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -92,7 +92,8 @@ def evaluate(
         temperature=temperature,
         top_p=top_p,
         top_k=top_k,
-        num_beams=num_beams
     )
     with torch.no_grad():
         generation_output = model.generate(

         temperature=temperature,
         top_p=top_p,
         top_k=top_k,
+        num_beams=num_beams,
+        **kwargs,
     )
     with torch.no_grad():
         generation_output = model.generate(