Unverified Commit 207ed8cb authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[s2s] use --eval_beams command line arg (#6926)

parent 0f360d3d
......@@ -202,6 +202,7 @@ class SummarizationModule(BaseTransformer):
attention_mask=batch["attention_mask"],
use_cache=True,
decoder_start_token_id=self.decoder_start_token_id,
num_beams=self.eval_beams,
)
gen_time = (time.time() - t0) / batch["input_ids"].shape[0]
preds: List[str] = self.ids_to_clean_text(generated_ids)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment