barreloflube
commited on
Commit
·
056aaff
1
Parent(s):
6324529
Refactor UI structure and add arbitrary types support
Browse files- src/tasks/images/sd.py +3 -0
src/tasks/images/sd.py
CHANGED
@@ -40,6 +40,9 @@ class ControlNetReq(BaseModel):
|
|
40 |
controlnets: List[str] # ["canny", "tile", "depth"]
|
41 |
control_images: List[Image.Image]
|
42 |
controlnet_conditioning_scale: List[float]
|
|
|
|
|
|
|
43 |
|
44 |
|
45 |
class SDReq(BaseModel):
|
|
|
40 |
controlnets: List[str] # ["canny", "tile", "depth"]
|
41 |
control_images: List[Image.Image]
|
42 |
controlnet_conditioning_scale: List[float]
|
43 |
+
|
44 |
+
class Config:
|
45 |
+
arbitrary_types_allowed=True
|
46 |
|
47 |
|
48 |
class SDReq(BaseModel):
|