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

feat(nodes): make ImageField properties required

Browse Source
This commit is contained in:
psychedelicious 2023-04-03 18:56:48 +10:00
parent d1058adb59
commit 5a9157e628
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/app/invocations/image.py
View File

@ -20,7 +20,14 @@ class ImageField(BaseModel):
) )
image_name: Optional[str] = Field(default=None, description="The name of the image") image_name: Optional[str] = Field(default=None, description="The name of the image")
class Config:
schema_extra = {
'required': [
'image_type',
'image_name',
]
}
class ImageOutput(BaseInvocationOutput): class ImageOutput(BaseInvocationOutput):
"""Base class for invocations that output an image""" """Base class for invocations that output an image"""
#fmt: off #fmt: off