diff --git a/tests/quantization/torchao/test_torchao.py b/tests/quantization/torchao/test_torchao.py index 38997de17b12..1b8caa481834 100644 --- a/tests/quantization/torchao/test_torchao.py +++ b/tests/quantization/torchao/test_torchao.py @@ -497,7 +497,7 @@ def test_memory_footprint(self): def test_model_memory_usage(self): model_id = "hf-internal-testing/tiny-flux-pipe" - expected_memory_saving_ratio = 2.0 + expected_memory_saving_ratio = 1.02 inputs = self.get_dummy_tensor_inputs(device=torch_device)