fix test compressed.py

neuralmagic · Jun 12, 2024 · cb285f4 · cb285f4
1 parent 041b5c1
commit cb285f4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/models/test_compressed.py b/tests/models/test_compressed.py
@@ -7,6 +7,7 @@
 """
 
 import pytest
+import gc
 
 from tests.models.utils import check_logprobs_close
 
@@ -40,6 +41,8 @@ def test_models(
         example_prompts, max_tokens, num_logprobs)
     del sparse_model
 
+    gc.collect()
+
     dense_model = vllm_runner(model_name=model_name,
                               sparsity=None,
                               dtype=dtype,