mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-09 00:09:45 +00:00
remove __class__.
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user