Update imports for example custom node code

This commit is contained in:
whjms 2023-11-20 23:52:26 -05:00 committed by GitHub
parent 487fda0226
commit 3e473ae008
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -65,7 +65,7 @@ The first set of things we need to do when creating a new Invocation are -
So let us do that. So let us do that.
```python ```python
from .baseinvocation import BaseInvocation, invocation from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
@invocation('resize') @invocation('resize')
class ResizeInvocation(BaseInvocation): class ResizeInvocation(BaseInvocation):
@ -99,8 +99,8 @@ create your own custom field types later in this guide. For now, let's go ahead
and use it. and use it.
```python ```python
from .baseinvocation import BaseInvocation, InputField, invocation from invokeai.app.invocations.baseinvocation import BaseInvocation, InputField, invocation
from .primitives import ImageField from invokeai.app.invocations.primitives import ImageField
@invocation('resize') @invocation('resize')
class ResizeInvocation(BaseInvocation): class ResizeInvocation(BaseInvocation):
@ -124,8 +124,8 @@ image: ImageField = InputField(description="The input image")
Great. Now let us create our other inputs for `width` and `height` Great. Now let us create our other inputs for `width` and `height`
```python ```python
from .baseinvocation import BaseInvocation, InputField, invocation from invokeai.app.invocations.baseinvocation import BaseInvocation, InputField, invocation
from .primitives import ImageField from invokeai.app.invocations.primitives import ImageField
@invocation('resize') @invocation('resize')
class ResizeInvocation(BaseInvocation): class ResizeInvocation(BaseInvocation):
@ -160,8 +160,8 @@ that are provided by it by InvokeAI.
Let us create this function first. Let us create this function first.
```python ```python
from .baseinvocation import BaseInvocation, InputField, invocation from invokeai.app.invocations.baseinvocation import BaseInvocation, InputField, invocation, InvocationContext
from .primitives import ImageField from invokeai.app.invocations.primitives import ImageField
@invocation('resize') @invocation('resize')
class ResizeInvocation(BaseInvocation): class ResizeInvocation(BaseInvocation):
@ -189,9 +189,9 @@ all the necessary info related to image outputs. So let us use that.
We will cover how to create your own output types later in this guide. We will cover how to create your own output types later in this guide.
```python ```python
from .baseinvocation import BaseInvocation, InputField, invocation from invokeai.app.invocations.baseinvocation import BaseInvocation, InputField, invocation, InvocationContext
from .primitives import ImageField from invokeai.app.invocations.primitives import ImageField
from .image import ImageOutput from invokeai.app.invocations.image import ImageOutput
@invocation('resize') @invocation('resize')
class ResizeInvocation(BaseInvocation): class ResizeInvocation(BaseInvocation):
@ -216,9 +216,9 @@ Perfect. Now that we have our Invocation setup, let us do what we want to do.
So let's do that. So let's do that.
```python ```python
from .baseinvocation import BaseInvocation, InputField, invocation from invokeai.app.invocations.baseinvocation import BaseInvocation, InputField, invocation, InvocationContext
from .primitives import ImageField from invokeai.app.invocations.primitives import ImageField
from .image import ImageOutput from invokeai.app.invocations.image import ImageOutput, ResourceOrigin, ImageCategory
@invocation("resize") @invocation("resize")
class ResizeInvocation(BaseInvocation): class ResizeInvocation(BaseInvocation):