storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove __class__.

Browse Source
This commit is contained in:
sayakpaul
2025-01-10 07:44:36 +05:30
parent 28b3e845d8
commit c0f38892f8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+1 -1
View File
@@ -471,7 +471,7 @@ class Trainer:
transformer_cls_ = unwrap_model(self.state.accelerator, self.transformer).__class__ transformer_cls_ = unwrap_model(self.state.accelerator, self.transformer).__class__
if self.args.training_type == "lora": 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" self.args.pretrained_model_name_or_path, subfolder="transformer"
) )
transformer_.add_adapter(transformer_lora_config) transformer_.add_adapter(transformer_lora_config)