SFTTrain error with code from website

If you copy paste the code from hugging face SFTTrainer page (Supervised Fine-tuning Trainer) it results in error:

name ‘PeftModel’ is not defined

Code:
trainer = SFTTrainer(
“facebook/opt-350m”,
train_dataset=dataset,
dataset_text_field=“text”,
max_seq_length=512,
)
trainer.train()