mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
c48fd9c083
Refine concept of "parameter" nodes to "primitives": - integer - float - string - boolean - image - latents - conditioning - color Each primitive has: - A field definition, if it is not already python primitive value. The field is how this primitive value is passed between nodes. Collections are lists of the field in node definitions. ex: `ImageField` & `list[ImageField]` - A single output class. ex: `ImageOutput` - A collection output class. ex: `ImageCollectionOutput` - A node, which functions to load or pass on the primitive value. ex: `ImageInvocation` (in this case, `ImageInvocation` replaces `LoadImage`) Plus a number of related changes: - Reorganize these into `primitives.py` - Update all nodes and logic to use primitives - Consolidate "prompt" outputs into "string" & "mask" into "image" (there's no reason for these to be different, the function identically) - Update default graphs & tests - Regen frontend types & minor frontend tidy related to changes
85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
# Copyright (c) 2023 Kyle Schouviller (https://github.com/kyle0654)
|
|
|
|
from typing import Literal
|
|
|
|
import numpy as np
|
|
|
|
from invokeai.app.invocations.primitives import IntegerOutput
|
|
|
|
from .baseinvocation import BaseInvocation, FieldDescriptions, InputField, InvocationContext, tags, title
|
|
|
|
|
|
@title("Add Integers")
|
|
@tags("math")
|
|
class AddInvocation(BaseInvocation):
|
|
"""Adds two numbers"""
|
|
|
|
type: Literal["add"] = "add"
|
|
|
|
# Inputs
|
|
a: int = InputField(default=0, description=FieldDescriptions.num_1)
|
|
b: int = InputField(default=0, description=FieldDescriptions.num_2)
|
|
|
|
def invoke(self, context: InvocationContext) -> IntegerOutput:
|
|
return IntegerOutput(a=self.a + self.b)
|
|
|
|
|
|
@title("Subtract Integers")
|
|
@tags("math")
|
|
class SubtractInvocation(BaseInvocation):
|
|
"""Subtracts two numbers"""
|
|
|
|
type: Literal["sub"] = "sub"
|
|
|
|
# Inputs
|
|
a: int = InputField(default=0, description=FieldDescriptions.num_1)
|
|
b: int = InputField(default=0, description=FieldDescriptions.num_2)
|
|
|
|
def invoke(self, context: InvocationContext) -> IntegerOutput:
|
|
return IntegerOutput(a=self.a - self.b)
|
|
|
|
|
|
@title("Multiply Integers")
|
|
@tags("math")
|
|
class MultiplyInvocation(BaseInvocation):
|
|
"""Multiplies two numbers"""
|
|
|
|
type: Literal["mul"] = "mul"
|
|
|
|
# Inputs
|
|
a: int = InputField(default=0, description=FieldDescriptions.num_1)
|
|
b: int = InputField(default=0, description=FieldDescriptions.num_2)
|
|
|
|
def invoke(self, context: InvocationContext) -> IntegerOutput:
|
|
return IntegerOutput(a=self.a * self.b)
|
|
|
|
|
|
@title("Divide Integers")
|
|
@tags("math")
|
|
class DivideInvocation(BaseInvocation):
|
|
"""Divides two numbers"""
|
|
|
|
type: Literal["div"] = "div"
|
|
|
|
# Inputs
|
|
a: int = InputField(default=0, description=FieldDescriptions.num_1)
|
|
b: int = InputField(default=0, description=FieldDescriptions.num_2)
|
|
|
|
def invoke(self, context: InvocationContext) -> IntegerOutput:
|
|
return IntegerOutput(a=int(self.a / self.b))
|
|
|
|
|
|
@title("Random Integer")
|
|
@tags("math")
|
|
class RandomIntInvocation(BaseInvocation):
|
|
"""Outputs a single random integer."""
|
|
|
|
type: Literal["rand_int"] = "rand_int"
|
|
|
|
# Inputs
|
|
low: int = InputField(default=0, description="The inclusive low value")
|
|
high: int = InputField(default=np.iinfo(np.int32).max, description="The exclusive high value")
|
|
|
|
def invoke(self, context: InvocationContext) -> IntegerOutput:
|
|
return IntegerOutput(a=np.random.randint(self.low, self.high))
|