apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat: split up pre-converted models into full and base sets

This commit is contained in:
Sean Sube 2023-12-23 21:36:52 -06:00
parent 83058f5a10
commit 75ac764d42
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
6 changed files with 304 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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...

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

@ -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...

151
models/preconverted-base-fp16.json Normal file
View File

@ -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"
}
]
}

151
models/preconverted-base-fp32.json Normal file
View File

@ -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"
}
]
}

0
models/preconverted-fp16.json → models/preconverted-full-fp16.json
View File

0
models/preconverted-fp32.json → models/preconverted-full-fp32.json
View File