From aac6b90d6163d52413e44cb3e1b379b4594e3a12 Mon Sep 17 00:00:00 2001 From: dbogunowicz <97082108+dbogunowicz@users.noreply.github.com> Date: Thu, 30 Nov 2023 11:37:16 +0100 Subject: [PATCH] [Cherry-Pick][Fix] Failing `test_save_zoo_directory` tests (#1869) * "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) * [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()