We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6308169 commit 54409faCopy full SHA for 54409fa
paddlenlp/prompt/prompt_trainer.py
@@ -366,7 +366,8 @@ def forward(self,
366
k: input_dict[k]
367
for k in input_dict if k in self.forward_keys
368
}
369
- model_inputs["masked_positions"] = None
+ if "masked_positions" in model_inputs:
370
+ model_inputs.pop("masked_positions")
371
outputs = self.plm(**model_inputs)
372
if self.verbalizer is not None:
373
label_outputs = self.verbalizer.process_outputs(
0 commit comments