diff --git a/openprompt/prompts/prefix_tuning_template.py b/openprompt/prompts/prefix_tuning_template.py index 9032307..14baca5 100644 --- a/openprompt/prompts/prefix_tuning_template.py +++ b/openprompt/prompts/prefix_tuning_template.py @@ -126,7 +126,7 @@ def get_past_key_values(self, batch_size=1): def generate_parameters(self) -> None: r""" - Generate parameters needed for new tokens' embedding in P-tuning + Generate parameters needed for new tokens' embedding in Prefix-tuning """ self.input_tokens = nn.Parameter(torch.arange(self.num_token).long(), requires_grad=False) # to allow automatic devicing