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 1
1
from typing import Literal , Optional
2
2
3
3
from PIL import Image
4
- import torchvision .transforms .functional as TVF
5
4
6
5
from invokeai .app .invocations .baseinvocation import (
7
- BaseModel ,
8
6
BaseInvocation ,
9
7
BaseInvocationOutput ,
10
8
Classification ,
@@ -62,19 +60,19 @@ class FluxUnoOutput(BaseInvocationOutput):
62
60
description = "Reference images container" , title = "Reference images"
63
61
)
64
62
65
- # TODO(attashe): adjust tags and category
63
+
66
64
@invocation (
67
65
"flux_uno" ,
68
66
title = "FLUX UNO" ,
69
- tags = ["ip_adapter " , "control" ],
67
+ tags = ["uno " , "control" ],
70
68
category = "ip_adapter" ,
71
69
version = "2.1.0" ,
72
70
classification = Classification .Beta ,
73
71
)
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 ."""
76
74
77
- image : ImageField = InputField (description = "The FLUX Redux image prompt ." )
75
+ image : ImageField = InputField (description = "The UNO reference image." )
78
76
image2 : Optional [ImageField ] = InputField (default = None , description = "2nd reference" )
79
77
image3 : Optional [ImageField ] = InputField (default = None , description = "3rd reference" )
80
78
image4 : Optional [ImageField ] = InputField (default = None , description = "4th reference" )
You can’t perform that action at this time.
0 commit comments