fix(exe): set default model cache to 0 for Windows bundle

ssube · Apr 22, 2023 · e89f2fc · e89f2fc
1 parent 21b53c1
commit e89f2fc
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/exe/onnx-web-full.bat b/exe/onnx-web-full.bat
@@ -1,8 +1,13 @@
+REM paths required to make the bundle work
 set ONNX_WEB_BASE_PATH=%~dp0
 set ONNX_WEB_BUNDLE_PATH=%ONNX_WEB_BASE_PATH%\client
 set ONNX_WEB_MODEL_PATH=%ONNX_WEB_BASE_PATH%\models
-set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\models\onnx-web-extras.json
 set ONNX_WEB_OUTPUT_PATH=%ONNX_WEB_BASE_PATH%\outputs
 
-@echo Launching onnx-web in full-precision mode...
+REM customize these as needed
+set ONNX_WEB_CACHE_MODELS=0
+set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\models\onnx-web-extras.json
+
+REM convert models and launch the server
+@echo Launching onnx-web in fp32 mode...
 server\onnx-web.exe --diffusion --correction --upscaling
diff --git a/exe/onnx-web-half.bat b/exe/onnx-web-half.bat
@@ -1,10 +1,14 @@
+REM paths required to make the bundle work
 set ONNX_WEB_BASE_PATH=%~dp0
 set ONNX_WEB_BUNDLE_PATH=%ONNX_WEB_BASE_PATH%\client
 set ONNX_WEB_MODEL_PATH=%ONNX_WEB_BASE_PATH%\models
-set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\models\onnx-web-extras.json
 set ONNX_WEB_OUTPUT_PATH=%ONNX_WEB_BASE_PATH%\outputs
 
+REM customize these as needed
 set ONNX_WEB_BLOCK_PLATFORMS=cpu
+set ONNX_WEB_CACHE_MODELS=0
+set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\models\onnx-web-extras.json
 
-@echo Launching onnx-web in half-precision mode...
+REM convert models and launch the server
+@echo Launching onnx-web in fp16 mode...
 server\onnx-web.exe --diffusion --correction --upscaling --half