barreloflube commited on
Commit
056aaff
·
1 Parent(s): 6324529

Refactor UI structure and add arbitrary types support

Browse files
Files changed (1) hide show
  1. 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):