barreloflube commited on
Commit
030e485
·
1 Parent(s): 42ae52a

Refactor ControlNetReq class to add controlnets, control_images, and controlnet_conditioning_scale attributes

Browse files
Files changed (1) hide show
  1. modules/helpers/common_helpers.py +9 -0
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 = ""