fix(api): move optimum model validation behind a feature flag, disabled by default
This commit is contained in:
parent
2d18ac1377
commit
bfdaae952f
|
@ -850,6 +850,7 @@ def convert_diffusion_diffusers_optimum(
|
||||||
"torch-fp16"
|
"torch-fp16"
|
||||||
), # optimum's fp16 mode only works on CUDA or ROCm
|
), # optimum's fp16 mode only works on CUDA or ROCm
|
||||||
framework="pt",
|
framework="pt",
|
||||||
|
do_validation=conversion.has_feature("optimum-validation"),
|
||||||
)
|
)
|
||||||
|
|
||||||
if "hash" in model:
|
if "hash" in model:
|
||||||
|
|
|
@ -94,6 +94,7 @@ def convert_diffusion_diffusers_xl(
|
||||||
"torch-fp16"
|
"torch-fp16"
|
||||||
), # optimum's fp16 mode only works on CUDA or ROCm
|
), # optimum's fp16 mode only works on CUDA or ROCm
|
||||||
framework="pt",
|
framework="pt",
|
||||||
|
do_validation=conversion.has_feature("optimum-validation"),
|
||||||
)
|
)
|
||||||
|
|
||||||
if "hash" in model:
|
if "hash" in model:
|
||||||
|
|
Loading…
Reference in New Issue