From f569bc682bb1998598f645ca9b5c98f847f5e90c Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Tue, 4 Apr 2023 10:21:38 +0200 Subject: [PATCH] Update src/peft/peft_model.py --- src/peft/peft_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py index 0305b79..6706b23 100644 --- a/src/peft/peft_model.py +++ b/src/peft/peft_model.py @@ -1066,7 +1066,7 @@ class PeftModelForVision2Seq(PeftModel): ... } >>> peft_config = get_peft_config(config) - >>> model = AutoModelForCausalLM.from_pretrained("Salesforce/blip2-flan-t5-xl") + >>> model = AutoModelForVision2Seq.from_pretrained("Salesforce/blip2-flan-t5-xl") >>> peft_model = PeftModelForVision2Seq(model, peft_config) >>> peft_model.print_trainable_parameters() trainable params: 1843200 || all params: 775873280 || trainable%: 0.23756456724479544