nict-astrec-att
/

prahokbart_base

Model card Files Files and versions

kainghour commited on Apr 29

Commit

1fbae43

·

verified ·

1 Parent(s): ed48588

beams=1

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -73,7 +73,7 @@ inp = tokenizer(
 ).input_ids
 model_output=model.generate(
-    inp, num_beams=4,
     decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<2km>")
 )

 ).input_ids
 model_output=model.generate(
+    inp,
     decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<2km>")
 )