From 17f51f0c92947a3ec9af15edf74fd5d577ee7cf1 Mon Sep 17 00:00:00 2001 From: Zero Rains Date: Mon, 4 Aug 2025 16:23:25 +0800 Subject: [PATCH] [unitest] fix the bug in test_sampler (#3157) --- test/layers/test_sampler.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/layers/test_sampler.py b/test/layers/test_sampler.py index 65a6bfbe6..c2fb69018 100644 --- 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