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 3
3
from invokeai .backend .bria .controlnet_bria import BriaControlModes , BriaMultiControlNetModel
4
4
from invokeai .backend .bria .controlnet_utils import prepare_control_images
5
5
from 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
7
7
8
8
import torch
9
9
from 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(
612
612
613
613
614
614
def 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 ,
623
623
):
624
624
# VAE applies 8x compression on images but we must also account for packing which requires
625
625
# latent height and width to be divisible by 2.
You can’t perform that action at this time.
0 commit comments