fix(api): make argument for skipping base models work correctly

ssube · Dec 23, 2023 · 50d51a3 · 50d51a3
1 parent c85b608
commit 50d51a3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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
+server\onnx-web.exe --no-base --diffusion --correction --upscaling
diff --git 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
+server\onnx-web.exe --no-base --diffusion --correction --upscaling --half