From 50d51a3921dfdc519b8799c318f4de6ce4c755be Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 23 Dec 2023 14:21:10 -0600 Subject: [PATCH] fix(api): make argument for skipping base models work correctly --- api/onnx_web/convert/__main__.py | 8 ++++---- exe/onnx-web-full.bat | 2 +- exe/onnx-web-half.bat | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py index 27d0c42d..89bf46c0 100644 --- a/api/onnx_web/convert/__main__.py +++ b/api/onnx_web/convert/__main__.py @@ -511,9 +511,9 @@ def main(args=None) -> int: ) # model groups - parser.add_argument("--base", action="store_true", default=True) - parser.add_argument("--networks", action="store_true", default=True) - parser.add_argument("--sources", action="store_true", default=True) + parser.add_argument("--no-base", action="store_true", default=False) + parser.add_argument("--networks", action="store_true", default=False) + parser.add_argument("--sources", action="store_true", default=False) parser.add_argument("--correction", action="store_true", default=False) parser.add_argument("--diffusion", action="store_true", default=False) parser.add_argument("--upscaling", action="store_true", default=False) @@ -568,7 +568,7 @@ def main(args=None) -> int: logger.info("model path does not existing, creating: %s", server.model_path) makedirs(server.model_path) - if args.base: + if not args.no_base: logger.info("converting base models") convert_models(server, args, base_models) diff --git a/exe/onnx-web-full.bat b/exe/onnx-web-full.bat index 9bba406b..9bfde595 100644 --- a/exe/onnx-web-full.bat +++ b/exe/onnx-web-full.bat @@ -10,4 +10,4 @@ set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_MODEL_PATH%\preconverted-fp32.json,%ONNX_WEB REM convert models and launch the server @echo Launching onnx-web in fp32 mode... -server\onnx-web.exe --base=false --diffusion --correction --upscaling \ No newline at end of file +server\onnx-web.exe --no-base --diffusion --correction --upscaling \ No newline at end of file diff --git a/exe/onnx-web-half.bat b/exe/onnx-web-half.bat index 109b8caa..3990b97f 100644 --- a/exe/onnx-web-half.bat +++ b/exe/onnx-web-half.bat @@ -11,4 +11,4 @@ set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_MODEL_PATH%\preconverted-fp16.json,%ONNX_WEB REM convert models and launch the server @echo Launching onnx-web in fp16 mode... -server\onnx-web.exe --base=false --diffusion --correction --upscaling --half \ No newline at end of file +server\onnx-web.exe --no-base --diffusion --correction --upscaling --half \ No newline at end of file