Skip to content

Commit 0977924

Browse files
Michael Gschwindfacebook-github-bot
authored andcommitted
Stop after first EOS post-promptfor FS2 (#2255)
Summary: Stop after first EOS post-promptfor FS2 Reviewed By: shoumikhin Differential Revision: D54525365
1 parent a1293b2 commit 0977924

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/models/llama2/builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ def _get_metadata(self):
231231
"get_max_batch_size": params.max_batch_size,
232232
"get_max_seq_len": params.max_seq_len,
233233
"get_n_bos": 1,
234-
"get_n_eos": 2 if is_fairseq2 else 1,
234+
"get_n_eos": 1, # 2 if is_fairseq2 else 1,
235235
"get_n_kv_heads": params.n_kv_heads,
236236
"get_n_layers": params.n_layers,
237237
"get_vocab_size": params.vocab_size,

0 commit comments

Comments
 (0)