Skip to content

Commit 37ebc28

Browse files
author
Ubuntu
committed
fixed node issue
1 parent c08a6a8 commit 37ebc28

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

invokeai/app/invocations/bria_denoiser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from invokeai.backend.bria.controlnet_bria import BriaControlModes, BriaMultiControlNetModel
44
from invokeai.backend.bria.controlnet_utils import prepare_control_images
55
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
77

88
import torch
99
from diffusers.schedulers.scheduling_flow_match_euler_discrete import FlowMatchEulerDiscreteScheduler

invokeai/backend/bria/pipeline_bria_controlnet.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -612,14 +612,14 @@ def encode_prompt(
612612

613613

614614
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,
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.

0 commit comments

Comments
 (0)