From c0f38892f88158d77fdeb2fc32d5eb3742a71473 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Fri, 10 Jan 2025 07:44:36 +0530 Subject: [PATCH] remove __class__. --- finetrainers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetrainers/trainer.py b/finetrainers/trainer.py index 922186d..e413fe8 100644 --- a/finetrainers/trainer.py +++ b/finetrainers/trainer.py @@ -471,7 +471,7 @@ class Trainer: transformer_cls_ = unwrap_model(self.state.accelerator, self.transformer).__class__ if self.args.training_type == "lora": - transformer_ = transformer_cls_.__class__.from_pretrained( + transformer_ = transformer_cls_.from_pretrained( self.args.pretrained_model_name_or_path, subfolder="transformer" ) transformer_.add_adapter(transformer_lora_config)