(
batch_size: int,
max_seq_len: int,
)
| 44 | |
| 45 | |
| 46 | def _create_tokens_tensor( |
| 47 | batch_size: int, |
| 48 | max_seq_len: int, |
| 49 | ) -> paddle.Tensor: |
| 50 | pre_token_ids = paddle.full(shape=[batch_size, max_seq_len], fill_value=-1, dtype="int64") |
| 51 | return pre_token_ids |
| 52 | |
| 53 | |
| 54 | def _create_default_sampling_metadata( |
no outgoing calls
no test coverage detected