[仅参数名不一致]transformers.StoppingCriteriaList

transformers.StoppingCriteriaList

transformers.StoppingCriteriaList(input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs)

paddlenlp.generation.StoppingCriteriaList

paddlenlp.generation.StoppingCriteriaList(input_ids: paddle.Tensor, scores: paddle.Tensor, **kwargs)

两者功能一致,仅参数名不一致,具体如下:

参数映射

transformers PaddlePaddle 备注
input_ids input_ids 输入(tokens) 的 id 组成的 Tensor。
scores scores 得分组成的 Tensor,仅参数名不一致。