Skip to content

Commit 8231878

Browse files
Attashehipsterusername
authored andcommitted
rename UNO node
1 parent c863ffe commit 8231878

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

invokeai/app/invocations/flux_uno.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,8 @@
11
from typing import Literal, Optional
22

33
from PIL import Image
4-
import torchvision.transforms.functional as TVF
54

65
from invokeai.app.invocations.baseinvocation import (
7-
BaseModel,
86
BaseInvocation,
97
BaseInvocationOutput,
108
Classification,
@@ -62,19 +60,19 @@ class FluxUnoOutput(BaseInvocationOutput):
6260
description="Reference images container", title="Reference images"
6361
)
6462

65-
# TODO(attashe): adjust tags and category
63+
6664
@invocation(
6765
"flux_uno",
6866
title="FLUX UNO",
69-
tags=["ip_adapter", "control"],
67+
tags=["uno", "control"],
7068
category="ip_adapter",
7169
version="2.1.0",
7270
classification=Classification.Beta,
7371
)
74-
class FluxReduxInvocation(BaseInvocation):
75-
"""Runs a FLUX Redux model to generate a conditioning tensor."""
72+
class FluxUnoInvocation(BaseInvocation):
73+
"""Loads a FLUX UNO reference images."""
7674

77-
image: ImageField = InputField(description="The FLUX Redux image prompt.")
75+
image: ImageField = InputField(description="The UNO reference image.")
7876
image2: Optional[ImageField] = InputField(default=None, description="2nd reference")
7977
image3: Optional[ImageField] = InputField(default=None, description="3rd reference")
8078
image4: Optional[ImageField] = InputField(default=None, description="4th reference")

0 commit comments

Comments
 (0)