File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -5693,10 +5693,13 @@ def load_lora_weights(
56935693        load_into_transformer_2  =  kwargs .pop ("load_into_transformer_2" , False )
56945694        if  load_into_transformer_2 :
56955695            if  not  hasattr (self , "transformer_2" ):
5696-                 raise  ValueError (
5697-                     "Cannot load LoRA into transformer_2: transformer_2 is not available for this model" 
5696+                 raise  AttributeError (
5697+                     f"'{ type (self ).__name__ }  
5698+                     "Note that Wan2.1 models do not have a transformer_2 component." 
56985699                    "Ensure the model has a transformer_2 component before setting load_into_transformer_2=True." 
56995700                )
5701+             if  "transformer_2"  not  in self ._lora_loadable_modules :
5702+                 self ._lora_loadable_modules .append ("transformer_2" )
57005703            self .load_lora_into_transformer (
57015704                state_dict ,
57025705                transformer = self .transformer_2 ,
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments