mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
23d65e7162
* Add latent to latent (img2img equivalent) Fix a CLI bug with multiple links per node * Using "latents" instead of "latent" * [nodes] In-progress implementation of graph library * Add linking to CLI for graph nodes (still broken) * Fix subgraph execution, fix subgraph linking in CLI * Fix LatentsToLatents
19 lines
595 B
Python
19 lines
595 B
Python
# Copyright (c) 2023 Kyle Schouviller (https://github.com/kyle0654)
|
|
|
|
from typing import Literal
|
|
from pydantic import Field
|
|
from .baseinvocation import BaseInvocation, BaseInvocationOutput, InvocationContext
|
|
from .math import IntOutput
|
|
|
|
# Pass-through parameter nodes - used by subgraphs
|
|
|
|
class ParamIntInvocation(BaseInvocation):
|
|
"""An integer parameter"""
|
|
#fmt: off
|
|
type: Literal["param_int"] = "param_int"
|
|
a: int = Field(default=0, description="The integer value")
|
|
#fmt: on
|
|
|
|
def invoke(self, context: InvocationContext) -> IntOutput:
|
|
return IntOutput(a=self.a)
|