diff --git a/exe/onnx-web-full.bat b/exe/onnx-web-full.bat index 9bfde595..29f89f8c 100644 --- a/exe/onnx-web-full.bat +++ b/exe/onnx-web-full.bat @@ -6,7 +6,7 @@ set ONNX_WEB_OUTPUT_PATH=%ONNX_WEB_BASE_PATH%\outputs REM customize these as needed set ONNX_WEB_CACHE_MODELS=0 -set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_MODEL_PATH%\preconverted-fp32.json,%ONNX_WEB_MODEL_PATH%\extras.json +set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_MODEL_PATH%\preconverted-base-fp32.json,%ONNX_WEB_MODEL_PATH%\extras.json REM convert models and launch the server @echo Launching onnx-web in fp32 mode... diff --git a/exe/onnx-web-half.bat b/exe/onnx-web-half.bat index 3990b97f..9eb6b230 100644 --- a/exe/onnx-web-half.bat +++ b/exe/onnx-web-half.bat @@ -7,7 +7,7 @@ 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_MODEL_PATH%\preconverted-fp16.json,%ONNX_WEB_MODEL_PATH%\extras.json +set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_MODEL_PATH%\preconverted-base-fp16.json,%ONNX_WEB_MODEL_PATH%\extras.json REM convert models and launch the server @echo Launching onnx-web in fp16 mode... diff --git a/models/preconverted-base-fp16.json b/models/preconverted-base-fp16.json new file mode 100644 index 00000000..cf957b65 --- /dev/null +++ b/models/preconverted-base-fp16.json @@ -0,0 +1,151 @@ +{ + "diffusion": [ + { + "format": "zip", + "label": "Dreamshaper v8 (fp16)", + "name": "diffusion-dreamshaper-v8-fp16", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/diffusion-dreamshaper-v8-fp16.zip" + }, + { + "format": "zip", + "label": "Stable Diffusion v1.5 (fp16)", + "name": "stable-diffusion-v1-5-fp16", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/stable-diffusion-v1-5-fp16.zip" + }, + { + "format": "zip", + "label": "SDXL - Turbo (fp16)", + "name": "stable-diffusion-xl-turbo-fp16", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/stable-diffusion-xl-turbo-fp16.zip" + }, + { + "format": "zip", + "label": "SDXL - Base v1.0 (fp16)", + "name": "stable-diffusion-xl-v1-0-fp16", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/stable-diffusion-xl-v1-0-fp16.zip" + } + ], + "sources": [ + { + "format": "onnx", + "name": "correction-gfpgan-v1-3", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/correction-gfpgan-v1-3.onnx" + }, + { + "format": "onnx", + "name": "correction-codeformer", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/correction-codeformer.onnx" + }, + { + "format": "onnx", + "name": "upscaling-real-esrgan-x2-plus", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-real-esrgan-x2-plus.onnx" + }, + { + "format": "onnx", + "name": "upscaling-real-esrgan-x4-plus", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-real-esrgan-x4-plus.onnx" + }, + { + "format": "onnx", + "name": "upscaling-real-esrgan-x4-v3", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-real-esrgan-x4-v3.onnx" + }, + { + "format": "onnx", + "name": "upscaling-swinir-classical-x4", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-swinir-classical-x4.onnx" + }, + { + "format": "onnx", + "name": "upscaling-swinir-real-large-x4", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-swinir-real-large-x4.onnx" + }, + { + "format": "onnx", + "name": "upscaling-bsrgan-x4", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-bsrgan-x4.onnx" + }, + { + "format": "onnx", + "name": "upscaling-bsrgan-x2", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-bsrgan-x2.onnx" + }, + { + "format": "pth", + "name": "detection-resnet50-final", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_Resnet50_Final.pth" + }, + { + "format": "pth", + "name": "detection-mobilenet025-final", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_mobilenet0.25_Final.pth" + }, + { + "format": "pth", + "name": "detection-yolo-v5-l", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5l-face.pth" + }, + { + "format": "pth", + "name": "detection-yolo-v5-n", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5n-face.pth" + }, + { + "format": "pth", + "name": "parsing-bisenet", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_bisenet.pth" + }, + { + "format": "pth", + "name": "parsing-parsenet", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_parsenet.pth" + }, + { + "dest": "control", + "format": "onnx", + "name": "canny", + "source": "https://huggingface.co/ForserX/sd-controlnet-canny-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "depth", + "source": "https://huggingface.co/ForserX/sd-controlnet-depth-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "hed", + "source": "https://huggingface.co/ForserX/sd-controlnet-hed-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "mlsd", + "source": "https://huggingface.co/ForserX/sd-controlnet-mlsd-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "normal", + "source": "https://huggingface.co/ForserX/sd-controlnet-normal-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "openpose", + "source": "https://huggingface.co/ForserX/sd-controlnet-openpose-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "seg", + "source": "https://huggingface.co/ForserX/sd-controlnet-seg-onnx/resolve/main/model.onnx" + } + ] +} \ No newline at end of file diff --git a/models/preconverted-base-fp32.json b/models/preconverted-base-fp32.json new file mode 100644 index 00000000..9a3bf99a --- /dev/null +++ b/models/preconverted-base-fp32.json @@ -0,0 +1,151 @@ +{ + "diffusion": [ + { + "format": "zip", + "label": "Dreamshaper v8 (fp32)", + "name": "diffusion-dreamshaper-v8-fp32", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/diffusion-dreamshaper-v8-fp32.zip" + }, + { + "format": "zip", + "label": "Stable Diffusion v1.5 (fp32)", + "name": "stable-diffusion-v1-5-fp32", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/stable-diffusion-v1-5-fp32.zip" + }, + { + "format": "zip", + "label": "SDXL - Turbo (fp32)", + "name": "stable-diffusion-xl-turbo-fp32", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/stable-diffusion-xl-turbo-fp32.zip" + }, + { + "format": "zip", + "label": "SDXL - Base v1.0 (fp32)", + "name": "stable-diffusion-xl-v1-0-fp32", + "pipeline": "archive", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/stable-diffusion-xl-v1-0-fp32.zip" + } + ], + "sources": [ + { + "format": "onnx", + "name": "correction-gfpgan-v1-3", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/correction-gfpgan-v1-3.onnx" + }, + { + "format": "onnx", + "name": "correction-codeformer", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/correction-codeformer.onnx" + }, + { + "format": "onnx", + "name": "upscaling-real-esrgan-x2-plus", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-real-esrgan-x2-plus.onnx" + }, + { + "format": "onnx", + "name": "upscaling-real-esrgan-x4-plus", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-real-esrgan-x4-plus.onnx" + }, + { + "format": "onnx", + "name": "upscaling-real-esrgan-x4-v3", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-real-esrgan-x4-v3.onnx" + }, + { + "format": "onnx", + "name": "upscaling-swinir-classical-x4", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-swinir-classical-x4.onnx" + }, + { + "format": "onnx", + "name": "upscaling-swinir-real-large-x4", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-swinir-real-large-x4.onnx" + }, + { + "format": "onnx", + "name": "upscaling-bsrgan-x4", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-bsrgan-x4.onnx" + }, + { + "format": "onnx", + "name": "upscaling-bsrgan-x2", + "source": "https://pub-ed4207c1d9d242ab9629ac1eb1f794cf.r2.dev/upscaling-bsrgan-x2.onnx" + }, + { + "format": "pth", + "name": "detection-resnet50-final", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_Resnet50_Final.pth" + }, + { + "format": "pth", + "name": "detection-mobilenet025-final", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/detection_mobilenet0.25_Final.pth" + }, + { + "format": "pth", + "name": "detection-yolo-v5-l", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5l-face.pth" + }, + { + "format": "pth", + "name": "detection-yolo-v5-n", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5n-face.pth" + }, + { + "format": "pth", + "name": "parsing-bisenet", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_bisenet.pth" + }, + { + "format": "pth", + "name": "parsing-parsenet", + "source": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_parsenet.pth" + }, + { + "dest": "control", + "format": "onnx", + "name": "canny", + "source": "https://huggingface.co/ForserX/sd-controlnet-canny-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "depth", + "source": "https://huggingface.co/ForserX/sd-controlnet-depth-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "hed", + "source": "https://huggingface.co/ForserX/sd-controlnet-hed-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "mlsd", + "source": "https://huggingface.co/ForserX/sd-controlnet-mlsd-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "normal", + "source": "https://huggingface.co/ForserX/sd-controlnet-normal-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "openpose", + "source": "https://huggingface.co/ForserX/sd-controlnet-openpose-onnx/resolve/main/model.onnx" + }, + { + "dest": "control", + "format": "onnx", + "name": "seg", + "source": "https://huggingface.co/ForserX/sd-controlnet-seg-onnx/resolve/main/model.onnx" + } + ] +} \ No newline at end of file diff --git a/models/preconverted-fp16.json b/models/preconverted-full-fp16.json similarity index 100% rename from models/preconverted-fp16.json rename to models/preconverted-full-fp16.json diff --git a/models/preconverted-fp32.json b/models/preconverted-full-fp32.json similarity index 100% rename from models/preconverted-fp32.json rename to models/preconverted-full-fp32.json