From 2737ec479421fb6f758bf1445afc049150c613fa Mon Sep 17 00:00:00 2001 From: dbogunowicz <97082108+dbogunowicz@users.noreply.github.com> Date: Wed, 22 Nov 2023 16:14:50 +0100 Subject: [PATCH 1/2] "AttributeError: 'torch._C.Value' object has no attribute 'float'" sparseml.transformers.export_onnx for zoo:bert-base_cased-squad_wikipedia_bookcorpus-pruned80.4block_quantized (#1847) --- src/sparseml/pytorch/utils/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sparseml/pytorch/utils/exporter.py b/src/sparseml/pytorch/utils/exporter.py index 4d7e44b51ac..d54875a98ea 100644 --- a/src/sparseml/pytorch/utils/exporter.py +++ b/src/sparseml/pytorch/utils/exporter.py @@ -460,7 +460,7 @@ def export_onnx( https://pytorch.org/docs/stable/onnx.html """ if _PARSED_TORCH_VERSION >= version.parse("1.10.0") and opset < 13 and convert_qat: - warnings.warn( + raise ValueError( "Exporting onnx with QAT and opset < 13 may result in errors. " "Please use opset>=13 with QAT. " "See https://github.com/pytorch/pytorch/issues/77455 for more info. " From c703ca2e8f07c8565b37e8cae1a70f84703324ba Mon Sep 17 00:00:00 2001 From: Damian Date: Thu, 30 Nov 2023 10:35:23 +0000 Subject: [PATCH 2/2] [Fix] Failing tests --- tests/sparseml/transformers/utils/test_helpers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/sparseml/transformers/utils/test_helpers.py b/tests/sparseml/transformers/utils/test_helpers.py index ae61578ccfe..ac668bdadc0 100644 --- a/tests/sparseml/transformers/utils/test_helpers.py +++ b/tests/sparseml/transformers/utils/test_helpers.py @@ -32,6 +32,7 @@ def test_save_zoo_directory(stub, tmp_path_factory): zoo_model = Model(stub, path_to_training_outputs) zoo_model.download() + zoo_model.deployment_tar.unzip() zoo_model.sample_inputs.unzip() zoo_model.sample_outputs["framework"].unzip()