File tree Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Original file line number Diff line number Diff line change 11from typing import Literal , Optional
22
33from PIL import Image
4- import torchvision .transforms .functional as TVF
54
65from 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" )
You can’t perform that action at this time.
0 commit comments