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