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: add missing primitive collections

Browse Source 
- add missing primitive collections
- remove `Seed` and `LoRAField` (they don't exist)
This commit is contained in:
psychedelicious
2023-08-15 22:18:37 +10:00
parent fa884134d9
commit 2b7dd3e236
8 changed files with 377 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
invokeai/app/invocations/collections.py
View File

@ -70,19 +70,3 @@ class RandomRangeInvocation(BaseInvocation):
def invoke(self, context: InvocationContext) -> IntegerCollectionOutput:
rng = np.random.default_rng(self.seed)
return IntegerCollectionOutput(collection=list(rng.integers(low=self.low, high=self.high, size=self.size)))
@title("Image Collection")
@tags("image", "collection")
class ImageCollectionInvocation(BaseInvocation):
"""Load a collection of images and provide it as output."""
type: Literal["image_collection"] = "image_collection"
# Inputs
images: list[ImageField] = InputField(
default=[], description="The image collection to load", ui_type=UIType.ImageCollection
)
def invoke(self, context: InvocationContext) -> ImageCollectionOutput:
return ImageCollectionOutput(collection=self.images)