fix img2img request spec

GooeyAI · Oct 3, 2023 · ed3c0cf · ed3c0cf
1 parent 131d8c7
commit ed3c0cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/recipes/Img2Img.py b/recipes/Img2Img.py
@@ -43,9 +43,9 @@ class RequestModel(BaseModel):
         text_prompt: str | None
 
         selected_model: typing.Literal[tuple(e.name for e in Img2ImgModels)] | None
-        selected_controlnet_model: typing.Literal[
-            tuple(e.name for e in ControlNetModels)
-        ] | None
+        selected_controlnet_model: list[
+            typing.Literal[tuple(e.name for e in ControlNetModels)]
+        ] | typing.Literal[tuple(e.name for e in ControlNetModels)] | None
         negative_prompt: str | None
 
         num_outputs: int | None