From bfdaae952f10be1c4770aeaebf9c59b86670de77 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 11 Feb 2024 16:57:05 -0600 Subject: [PATCH] fix(api): move optimum model validation behind a feature flag, disabled by default --- api/onnx_web/convert/diffusion/diffusion.py | 1 + api/onnx_web/convert/diffusion/diffusion_xl.py | 1 + 2 files changed, 2 insertions(+) diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py index 969d0ea2..fb72b43f 100644 --- a/api/onnx_web/convert/diffusion/diffusion.py +++ b/api/onnx_web/convert/diffusion/diffusion.py @@ -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: diff --git a/api/onnx_web/convert/diffusion/diffusion_xl.py b/api/onnx_web/convert/diffusion/diffusion_xl.py index 59708fa7..5d88e1a5 100644 --- a/api/onnx_web/convert/diffusion/diffusion_xl.py +++ b/api/onnx_web/convert/diffusion/diffusion_xl.py @@ -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: