File tree Expand file tree Collapse file tree 2 files changed +9
-9
lines changed Expand file tree Collapse file tree 2 files changed +9
-9
lines changed Original file line number Diff line number Diff line change 33from  invokeai .backend .bria .controlnet_bria  import  BriaControlModes , BriaMultiControlNetModel 
44from  invokeai .backend .bria .controlnet_utils  import  prepare_control_images 
55from  invokeai .backend .bria .pipeline_bria_controlnet  import  BriaControlNetPipeline 
6- from  invokeai .nodes . bria_nodes .bria_controlnet  import  BriaControlNetField 
6+ from  invokeai .app . invocations .bria_controlnet  import  BriaControlNetField 
77
88import  torch 
99from  diffusers .schedulers .scheduling_flow_match_euler_discrete  import  FlowMatchEulerDiscreteScheduler 
Original file line number Diff line number Diff line change @@ -612,14 +612,14 @@ def encode_prompt(
612612
613613
614614def  prepare_latents (
615-     batch_size ,
616-     num_channels_latents ,
617-     height ,
618-     width ,
619-     dtype ,
620-     device ,
621-     generator ,
622-     latents = None ,
615+     batch_size :  int ,
616+     num_channels_latents :  int ,
617+     height :  int ,
618+     width :  int ,
619+     dtype :  torch . dtype ,
620+     device :  torch . device ,
621+     generator :  torch . Generator ,
622+     latents :  Optional [ torch . FloatTensor ]  =   None ,
623623):
624624    # VAE applies 8x compression on images but we must also account for packing which requires 
625625    # latent height and width to be divisible by 2. 
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments