Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactored ControlNet nodes so they subclass from PreprocessedControlInvocation, and only need to override run_processor(image) (instead of reimplementing invoke())

Browse Source
This commit is contained in:
user1 2023-05-04 16:01:22 -07:00 committed by Kent Keirsey
parent 4ae4bf4ff9
commit 6d958a214c
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/app/invocations/controlnet_image_processors.py
View File

@ -113,7 +113,10 @@ class ControlOutput(BaseInvocationOutput):
control: ControlField = Field(default=None, description="The control info dict")
# fmt: on
class PreprocessedControlInvocation(BaseInvocation, PILInvocationConfig):
"""Base class for invocations that preprocess images for ControlNet"""
<<<<<<< HEAD
class ControlNetInvocation(BaseInvocation):
"""Collects ControlNet info to pass to other nodes"""
# fmt: off