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

fix: lint erros

Browse Source
This commit is contained in:
blessedcoolant 2024-02-10 04:02:05 +05:30 committed by Kent Keirsey
parent 7d80261d47
commit 67daf1751c
1 changed files with 30 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,30 +6,45 @@ from typing import Dict, List, Literal, Union
import cv2 import cv2
import numpy as np import numpy as np
from controlnet_aux import (CannyDetector, ContentShuffleDetector, HEDdetector, from controlnet_aux import (
LeresDetector, LineartAnimeDetector, CannyDetector,
LineartDetector, MediapipeFaceDetector, ContentShuffleDetector,
MidasDetector, MLSDdetector, NormalBaeDetector, HEDdetector,
OpenposeDetector, PidiNetDetector, SamDetector, LeresDetector,
ZoeDetector) LineartAnimeDetector,
LineartDetector,
MediapipeFaceDetector,
MidasDetector,
MLSDdetector,
NormalBaeDetector,
OpenposeDetector,
PidiNetDetector,
SamDetector,
ZoeDetector,
)
from controlnet_aux.util import HWC3, ade_palette from controlnet_aux.util import HWC3, ade_palette
from PIL import Image from PIL import Image
from pydantic import (BaseModel, ConfigDict, Field, field_validator, from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
model_validator)
from invokeai.app.invocations.primitives import ImageField, ImageOutput from invokeai.app.invocations.primitives import ImageField, ImageOutput
from invokeai.app.invocations.util import (validate_begin_end_step, from invokeai.app.invocations.util import validate_begin_end_step, validate_weights
validate_weights) from invokeai.app.services.image_records.image_records_common import ImageCategory, ResourceOrigin
from invokeai.app.services.image_records.image_records_common import (
ImageCategory, ResourceOrigin)
from invokeai.app.shared.fields import FieldDescriptions from invokeai.app.shared.fields import FieldDescriptions
from invokeai.backend.image_util.depth_anything import DepthAnythingDetector from invokeai.backend.image_util.depth_anything import DepthAnythingDetector
from invokeai.backend.image_util.dwpose import DWPoseDetector from invokeai.backend.image_util.dwpose import DWPoseDetector
from ...backend.model_management import BaseModelType from ...backend.model_management import BaseModelType
from .baseinvocation import (BaseInvocation, BaseInvocationOutput, Input, from .baseinvocation import (
InputField, InvocationContext, OutputField, BaseInvocation,
WithMetadata, invocation, invocation_output) BaseInvocationOutput,
Input,
InputField,
InvocationContext,
OutputField,
WithMetadata,
invocation,
invocation_output,
)
CONTROLNET_MODE_VALUES = Literal["balanced", "more_prompt", "more_control", "unbalanced"] CONTROLNET_MODE_VALUES = Literal["balanced", "more_prompt", "more_control", "unbalanced"]
CONTROLNET_RESIZE_VALUES = Literal[ CONTROLNET_RESIZE_VALUES = Literal[