feat(nodes): create invocation_api.py

This is the public API for invocations.

Everything a custom node might need should be re-exported from this file.
This commit is contained in:
psychedelicious
2024-01-15 10:48:33 +11:00
parent 05fb485d33
commit a466f7a94b
7 changed files with 116 additions and 9 deletions

View File

@ -25,8 +25,7 @@ from controlnet_aux.util import HWC3, ade_palette
from PIL import Image
from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
from invokeai.app.invocations.baseinvocation import WithMetadata
from invokeai.app.invocations.fields import FieldDescriptions, ImageField, Input, InputField, OutputField
from invokeai.app.invocations.fields import FieldDescriptions, ImageField, Input, InputField, OutputField, WithMetadata
from invokeai.app.invocations.primitives import ImageOutput
from invokeai.app.invocations.util import validate_begin_end_step, validate_weights
from invokeai.backend.image_util.depth_anything import DepthAnythingDetector

View File

@ -13,11 +13,10 @@ from pydantic import field_validator
import invokeai.assets.fonts as font_assets
from invokeai.app.invocations.baseinvocation import (
BaseInvocation,
WithMetadata,
invocation,
invocation_output,
)
from invokeai.app.invocations.fields import ImageField, InputField, OutputField
from invokeai.app.invocations.fields import ImageField, InputField, OutputField, WithMetadata
from invokeai.app.invocations.primitives import ImageOutput
from invokeai.app.services.image_records.image_records_common import ImageCategory

View File

@ -7,7 +7,6 @@ import cv2
import numpy
from PIL import Image, ImageChops, ImageFilter, ImageOps
from invokeai.app.invocations.baseinvocation import WithMetadata
from invokeai.app.invocations.fields import (
BoardField,
ColorField,
@ -15,6 +14,7 @@ from invokeai.app.invocations.fields import (
ImageField,
Input,
InputField,
WithMetadata,
)
from invokeai.app.invocations.primitives import ImageOutput
from invokeai.app.services.image_records.image_records_common import ImageCategory

View File

@ -13,8 +13,8 @@ from invokeai.backend.image_util.cv2_inpaint import cv2_inpaint
from invokeai.backend.image_util.lama import LaMA
from invokeai.backend.image_util.patchmatch import PatchMatch
from .baseinvocation import BaseInvocation, WithMetadata, invocation
from .fields import InputField
from .baseinvocation import BaseInvocation, invocation
from .fields import InputField, WithMetadata
from .image import PIL_RESAMPLING_MAP, PIL_RESAMPLING_MODES

View File

@ -8,11 +8,10 @@ from invokeai.app.invocations.baseinvocation import (
BaseInvocation,
BaseInvocationOutput,
Classification,
WithMetadata,
invocation,
invocation_output,
)
from invokeai.app.invocations.fields import ImageField, Input, InputField, OutputField
from invokeai.app.invocations.fields import ImageField, Input, InputField, OutputField, WithMetadata
from invokeai.app.invocations.primitives import ImageOutput
from invokeai.backend.tiles.tiles import (
calc_tiles_even_split,