From 5d570064a40941f32ac4b0821dc4fdd90f6c5a0a Mon Sep 17 00:00:00 2001 From: adroitHou Date: Wed, 30 Aug 2023 10:24:16 -0400 Subject: [PATCH] FIX: PeftModel save_pretrained Doc (#881) --- 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 f7d6758508..20d444a377 100644 --- a/src/peft/peft_model.py +++ b/src/peft/peft_model.py @@ -134,7 +134,7 @@ def save_pretrained( ): r""" This function saves the adapter model and the adapter configuration files to a directory, so that it can be - reloaded using the [`LoraModel.from_pretrained`] class method, and also used by the [`LoraModel.push_to_hub`] + reloaded using the [`PeftModel.from_pretrained`] class method, and also used by the [`PeftModel.push_to_hub`] method. Args: