fix(api): make LPW pipeline valid again

ssube · Jun 11, 2023 · d47dab8 · d47dab8
1 parent 82a80bb
commit d47dab8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py
@@ -250,10 +250,10 @@ def get_valid_pipeline(self, group: str, pipeline: str = None) -> str:
             if pipeline in ["controlnet", "lpw", "panorama", "pix2pix"]:
                 return pipeline
         elif group == "inpaint":
-            if pipeline in ["controlnet", "panorama"]:
+            if pipeline in ["controlnet", "lpw", "panorama"]:
                 return pipeline
         elif group == "txt2img":
-            if pipeline in ["panorama"]:
+            if pipeline in ["lpw", "panorama"]:
                 return pipeline
 
         logger.debug("pipeline %s is not valid for %s", pipeline, group)