diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py index 28a319fb..0b68ecc1 100644 --- a/api/onnx_web/convert/__main__.py +++ b/api/onnx_web/convert/__main__.py @@ -546,6 +546,14 @@ def main(args=None) -> int: server.opset = args.opset server.token = args.token + # debug options + if server.debug: + import debugpy + + debugpy.listen(5678) + logger.warning("waiting for debugger") + debugpy.wait_for_client() + register_plugins(server) logger.info( diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py index a2e3d807..2587440b 100644 --- a/api/onnx_web/convert/diffusion/diffusion.py +++ b/api/onnx_web/convert/diffusion/diffusion.py @@ -507,7 +507,7 @@ def convert_diffusion_diffusers( run_gc() if conversion.control and not single_vae and not conversion.share_unet: - cnet_source = torch_source or source + cnet_source = torch_source or cache_path logger.info("loading and converting CNet from %s", cnet_source) cnet_path = convert_diffusion_diffusers_cnet( conversion, diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index 7f28523b..a5533cf7 100644 --- a/api/onnx_web/diffusers/load.py +++ b/api/onnx_web/diffusers/load.py @@ -235,7 +235,8 @@ def load_pipeline( pipeline_class.__name__, ) pipe = pipeline_class( - components["vae"], + components["vae_encoder"], + components["vae_decoder"], components["text_encoder"], components["tokenizer"], components["unet"], diff --git a/api/onnx_web/image/source_filter.py b/api/onnx_web/image/source_filter.py index 99c6b5eb..9ad9237f 100644 --- a/api/onnx_web/image/source_filter.py +++ b/api/onnx_web/image/source_filter.py @@ -160,7 +160,7 @@ def source_filter_hed(server: ServerContext, source: Image.Image) -> Image.Image logger.debug("running HED detection on source image") hed = HEDdetector.from_pretrained( - "lllyasviel/ControlNet", + "lllyasviel/Annotators", cache_dir=server.cache_path, ) image = hed(source) @@ -172,7 +172,7 @@ def source_filter_scribble(server: ServerContext, source: Image.Image) -> Image. logger.debug("running scribble detection on source image") hed = HEDdetector.from_pretrained( - "lllyasviel/ControlNet", + "lllyasviel/Annotators", cache_dir=server.cache_path, ) image = hed(source, scribble=True)