apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-12-23 14:21:10 -06:00
parent c85b608196
commit 50d51a3921
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/convert/__main__.py
View File

@ -511,9 +511,9 @@ def main(args=None) -> int:
) )
# model groups # model groups
parser.add_argument("--base", action="store_true", default=True) parser.add_argument("--no-base", action="store_true", default=False)
parser.add_argument("--networks", action="store_true", default=True) parser.add_argument("--networks", action="store_true", default=False)
parser.add_argument("--sources", action="store_true", default=True) parser.add_argument("--sources", action="store_true", default=False)
parser.add_argument("--correction", 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("--diffusion", action="store_true", default=False)
parser.add_argument("--upscaling", 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) logger.info("model path does not existing, creating: %s", server.model_path)
makedirs(server.model_path) makedirs(server.model_path)
if args.base: if not args.no_base:
logger.info("converting base models") logger.info("converting base models")
convert_models(server, args, base_models) convert_models(server, args, base_models)

2
exe/onnx-web-full.bat
View File

@ -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 REM convert models and launch the server
@echo Launching onnx-web in fp32 mode... @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

2
exe/onnx-web-half.bat
View File

@ -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 REM convert models and launch the server
@echo Launching onnx-web in fp16 mode... @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