File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -877,15 +877,11 @@ def test_simple_inference_with_text_denoiser_lora_unfused(
877877            pipe , denoiser  =  self .check_if_adapters_added_correctly (pipe , text_lora_config , denoiser_lora_config )
878878
879879            pipe .fuse_lora (components = self .pipeline_class ._lora_loadable_modules )
880-             self .assertTrue (
881-                 pipe .num_fused_loras  ==  1 , pipe .num_fused_loras , f"{ pipe .num_fused_loras = } { pipe .fused_loras = }  
882-             )
880+             self .assertTrue (pipe .num_fused_loras  ==  1 , f"{ pipe .num_fused_loras = } { pipe .fused_loras = }  )
883881            output_fused_lora  =  pipe (** inputs , generator = torch .manual_seed (0 ))[0 ]
884882
885883            pipe .unfuse_lora (components = self .pipeline_class ._lora_loadable_modules )
886-             self .assertTrue (
887-                 pipe .num_fused_loras  ==  0 , pipe .num_fused_loras , f"{ pipe .num_fused_loras = } { pipe .fused_loras = }  
888-             )
884+             self .assertTrue (pipe .num_fused_loras  ==  0 , f"{ pipe .num_fused_loras = } { pipe .fused_loras = }  )
889885            output_unfused_lora  =  pipe (** inputs , generator = torch .manual_seed (0 ))[0 ]
890886
891887            # unloading should remove the LoRA layers 
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments