apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): move optimum model validation behind a feature flag, disabled by default

This commit is contained in:
Sean Sube 2024-02-11 16:57:05 -06:00
parent 2d18ac1377
commit bfdaae952f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/convert/diffusion/diffusion.py
View File

@ -850,6 +850,7 @@ def convert_diffusion_diffusers_optimum(
"torch-fp16"
), # optimum's fp16 mode only works on CUDA or ROCm
framework="pt",
do_validation=conversion.has_feature("optimum-validation"),
)
if "hash" in model:

1
api/onnx_web/convert/diffusion/diffusion_xl.py
View File

@ -94,6 +94,7 @@ def convert_diffusion_diffusers_xl(
"torch-fp16"
), # optimum's fp16 mode only works on CUDA or ROCm
framework="pt",
do_validation=conversion.has_feature("optimum-validation"),
)
if "hash" in model: