DOC fix wrong import in p-tuning docs (#1049)

huggingface · Oct 25, 2023 · 8b21a4e · 8b21a4e
1 parent 894e68a
commit 8b21a4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/task_guides/ptuning-seq-classification.mdx b/docs/source/task_guides/ptuning-seq-classification.mdx
@@ -197,7 +197,7 @@ Once the model has been uploaded to the Hub, anyone can easily use it for infere
 ```py
 import torch
 from peft import PeftModel, PeftConfig
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 peft_model_id = "smangrul/roberta-large-peft-p-tuning"
 config = PeftConfig.from_pretrained(peft_model_id)