[unitest] fix the bug in test_sampler (#3157)

2025-10-05 16:48:03 +08:00 · 2025-08-04 16:23:25 +08:00
parent 79bbacc152
commit 17f51f0c92
1 changed files with 1 additions and 0 deletions
--- a/test/layers/test_sampler.py
+++ b/test/layers/test_sampler.py
@@ -56,6 +56,7 @@ def _create_default_sampling_metadata(
        min_dec_lens=paddle.full(shape=[batch_size, 1], fill_value=min_seq_len, dtype="int64"),
        bad_words_token_ids=paddle.full(shape=[batch_size], fill_value=-1, dtype="int64"),
        eos_token_ids=paddle.full(shape=[batch_size], fill_value=-2, dtype="int64"),
+        min_p=paddle.randn([batch_size]),
    )
    return fake_sampling_metadata