From b80e334e719de76ca47b964b06b62abf7144cddc Mon Sep 17 00:00:00 2001 From: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com> Date: Mon, 20 Jan 2025 15:00:34 +0100 Subject: [PATCH] Skip Falcon 7B GGML Test (#35783) skip test --- tests/quantization/ggml/test_ggml.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/quantization/ggml/test_ggml.py b/tests/quantization/ggml/test_ggml.py index 1cd9c2d4a8..ad5cdb17fe 100644 --- a/tests/quantization/ggml/test_ggml.py +++ b/tests/quantization/ggml/test_ggml.py @@ -636,6 +636,7 @@ class GgufIntegrationTests(unittest.TestCase): EXPECTED_TEXT = 'Hello,\nI am trying to use the "get_post_meta"' self.assertEqual(tokenizer.decode(out[0], skip_special_tokens=True), EXPECTED_TEXT) + @unittest.skip("The test causes a torch.OutOfMemoryError on the CI but it passes with enough memory") def test_falcon7b_weights_conversion_fp16(self): quantized_model = AutoModelForCausalLM.from_pretrained( self.falcon7b_model_id_fp16,