diff --git a/modelscope/models/nlp/space/model/generator.py b/modelscope/models/nlp/space/model/generator.py index fab70fd6..08e1c765 100644 --- a/modelscope/models/nlp/space/model/generator.py +++ b/modelscope/models/nlp/space/model/generator.py @@ -183,7 +183,8 @@ class BeamSearch(Generator): scores_after_end = np.full(self.vocab_size, -1e10, dtype='float32') scores_after_end[ - self.pad_id] = 0 # we want is generated after ,so maximum log(p()) is (0) + self. + pad_id] = 0 # we want is generated after ,so maximum log(p()) is (0) scores_after_end = torch.from_numpy(scores_after_end) if self.use_gpu: