File tree Expand file tree Collapse file tree 1 file changed +7
-11
lines changed
Expand file tree Collapse file tree 1 file changed +7
-11
lines changed Original file line number Diff line number Diff line change @@ -2401,17 +2401,13 @@ def load_lora_into_transformer(
24012401 )
24022402
24032403 # Load the layers corresponding to transformer.
2404- keys = list (state_dict .keys ())
2405- transformer_present = any (key .startswith (cls .transformer_name ) for key in keys )
2406- if transformer_present :
2407- logger .info (f"Loading { cls .transformer_name } ." )
2408- transformer .load_lora_adapter (
2409- state_dict ,
2410- network_alphas = network_alphas ,
2411- adapter_name = adapter_name ,
2412- _pipeline = _pipeline ,
2413- low_cpu_mem_usage = low_cpu_mem_usage ,
2414- )
2404+ transformer .load_lora_adapter (
2405+ state_dict ,
2406+ network_alphas = network_alphas ,
2407+ adapter_name = adapter_name ,
2408+ _pipeline = _pipeline ,
2409+ low_cpu_mem_usage = low_cpu_mem_usage ,
2410+ )
24152411
24162412 @classmethod
24172413 # Copied from diffusers.loaders.lora_pipeline.StableDiffusionLoraLoaderMixin.load_lora_into_text_encoder
You can’t perform that action at this time.
0 commit comments