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