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

Commented out ZoeDetector. Will re-instate once there's a controlnet-aux release that supports it.

Browse Source
This commit is contained in:
user1 2023-05-13 04:17:53 -07:00 committed by Kent Keirsey
parent 9ec3d2bc0c
commit 51032dc0b2
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -26,7 +26,8 @@ from controlnet_aux import (
OpenposeDetector, OpenposeDetector,
PidiNetDetector, PidiNetDetector,
ContentShuffleDetector, ContentShuffleDetector,
ZoeDetector) # ZoeDetector, # FIXME: uncomment once ZoeDetector is availabel in official controlnet_aux release
)
from .image import ImageOutput, build_image_output, PILInvocationConfig from .image import ImageOutput, build_image_output, PILInvocationConfig
@ -324,14 +325,16 @@ class ContentShuffleImageProcessorInvocation(ImageProcessorInvocation, PILInvoca
return processed_image return processed_image
class ZoeDepthImageProcessorInvocation(ImageProcessorInvocation, PILInvocationConfig): # # FIXME: ZoeDetector was implemented _after_ most recent official release of controlnet_aux (v0.0.3)
"""Applies Zoe depth processing to image""" # # so it is commented out until a new release is made
# fmt: off # class ZoeDepthImageProcessorInvocation(ImageProcessorInvocation, PILInvocationConfig):
type: Literal["zoe_depth_image_processor"] = "zoe_depth_image_processor" # """Applies Zoe depth processing to image"""
# fmt: on # # fmt: off
# type: Literal["zoe_depth_image_processor"] = "zoe_depth_image_processor"
def run_processor(self, image): # # fmt: on
zoe_depth_processor = ZoeDetector.from_pretrained("lllyasviel/Annotators") #
processed_image = zoe_depth_processor(image) # def run_processor(self, image):
return processed_image # zoe_depth_processor = ZoeDetector.from_pretrained("lllyasviel/Annotators")
# processed_image = zoe_depth_processor(image)
# return processed_image