Commit
·
030e485
1
Parent(s):
42ae52a
Refactor ControlNetReq class to add controlnets, control_images, and controlnet_conditioning_scale attributes
Browse files
modules/helpers/common_helpers.py
CHANGED
@@ -10,6 +10,15 @@ from controlnet_aux.processor import Processor
|
|
10 |
from .flux_helpers import ControlNetReq
|
11 |
|
12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
class BaseReq(BaseModel):
|
14 |
model: str = ""
|
15 |
prompt: str = ""
|
|
|
10 |
from .flux_helpers import ControlNetReq
|
11 |
|
12 |
|
13 |
+
class ControlNetReq(BaseModel):
|
14 |
+
controlnets: List[str] # ["canny", "tile", "depth"]
|
15 |
+
control_images: List[Image.Image]
|
16 |
+
controlnet_conditioning_scale: List[float]
|
17 |
+
|
18 |
+
class Config:
|
19 |
+
arbitrary_types_allowed=True
|
20 |
+
|
21 |
+
|
22 |
class BaseReq(BaseModel):
|
23 |
model: str = ""
|
24 |
prompt: str = ""
|