diff --git a/peft_docs/en/clm-prompt-tuning.ipynb b/peft_docs/en/clm-prompt-tuning.ipynb index 205ccc51..c235d2a1 100644 --- a/peft_docs/en/clm-prompt-tuning.ipynb +++ b/peft_docs/en/clm-prompt-tuning.ipynb @@ -197,7 +197,7 @@ " model_inputs[\"attention_mask\"][i] = [0] * (max_length - len(sample_input_ids)) + model_inputs[\n", " \"attention_mask\"\n", " ][i]\n", - " labels[\"input_ids\"][i] = [-100] * (max_length - len(sample_input_ids)) + label_input_ids\n", + " labels[\"input_ids\"][i] = [-100] * (max_length - len(label_input_ids)) + label_input_ids\n", " model_inputs[\"input_ids\"][i] = torch.tensor(model_inputs[\"input_ids\"][i][:max_length])\n", " model_inputs[\"attention_mask\"][i] = torch.tensor(model_inputs[\"attention_mask\"][i][:max_length])\n", " labels[\"input_ids\"][i] = torch.tensor(labels[\"input_ids\"][i][:max_length])\n",