github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Create a dedicated FLUX ControlNet invocation.

Browse Source
This commit is contained in:
Ryan Dick
2024-10-08 21:24:55 +00:00
parent 0dd9f1f772
commit dea6cbd599
4 changed files with 104 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/backend/flux/extensions/instantx_controlnet_extension.py
View File

@@ -5,7 +5,7 @@ from PIL.Image import Image
from invokeai.app.invocations.constants import LATENT_SCALE_FACTOR
from invokeai.app.invocations.flux_vae_encode import FluxVaeEncodeInvocation
from invokeai.app.util.controlnet_utils import CONTROLNET_MODE_VALUES, CONTROLNET_RESIZE_VALUES, prepare_control_image
from invokeai.app.util.controlnet_utils import CONTROLNET_RESIZE_VALUES, prepare_control_image
from invokeai.backend.flux.controlnet.instantx_controlnet_flux import (
InstantXControlNetFlux,
)
@@ -51,7 +51,6 @@ class InstantXControlNetExtension(BaseControlNetExtension):
latent_width: int,
dtype: torch.dtype,
device: torch.device,
control_mode: CONTROLNET_MODE_VALUES,
resize_mode: CONTROLNET_RESIZE_VALUES,
weight: Union[float, List[float]],
begin_step_percent: float,
@@ -67,7 +66,7 @@ class InstantXControlNetExtension(BaseControlNetExtension):
height=image_height,
device=device,
dtype=dtype,
control_mode=control_mode,
control_mode="balanced",
resize_mode=resize_mode,
)

5
invokeai/backend/flux/extensions/xlabs_controlnet_extension.py
View File

@@ -4,7 +4,7 @@ import torch
from PIL.Image import Image
from invokeai.app.invocations.constants import LATENT_SCALE_FACTOR
from invokeai.app.util.controlnet_utils import CONTROLNET_MODE_VALUES, CONTROLNET_RESIZE_VALUES, prepare_control_image
from invokeai.app.util.controlnet_utils import CONTROLNET_RESIZE_VALUES, prepare_control_image
from invokeai.backend.flux.controlnet.xlabs_controlnet_flux import XLabsControlNetFlux
from invokeai.backend.flux.controlnet.xlabs_controlnet_flux_output import XLabsControlNetFluxOutput
from invokeai.backend.flux.extensions.base_controlnet_extension import BaseControlNetExtension
@@ -39,7 +39,6 @@ class XLabsControlNetExtension(BaseControlNetExtension):
latent_width: int,
dtype: torch.dtype,
device: torch.device,
control_mode: CONTROLNET_MODE_VALUES,
resize_mode: CONTROLNET_RESIZE_VALUES,
weight: Union[float, List[float]],
begin_step_percent: float,
@@ -55,7 +54,7 @@ class XLabsControlNetExtension(BaseControlNetExtension):
height=image_height,
device=device,
dtype=dtype,
control_mode=control_mode,
control_mode="balanced",
resize_mode=resize_mode,
)