Skip Falcon 7B GGML Test (#35783)

skip test
huggingface · Jan 20, 2025 · b80e334 · b80e334
1 parent 6894728
commit b80e334
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/quantization/ggml/test_ggml.py b/tests/quantization/ggml/test_ggml.py
@@ -636,6 +636,7 @@ def test_falcon7b_q2_k(self):
         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,