tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix: Test

Browse Source 
Signed-off-by: Rahul Tuli <rtuli@redhat.com>
This commit is contained in:
Rahul Tuli
2025-04-28 09:18:26 -04:00
committed by Wing Lian
parent f4bcc71c86
commit 5246aebc04
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/e2e/integrations/test_llm_compressor.py
View File

@@ -90,12 +90,9 @@ class TestLLMCompressorIntegration:
def _check_llmcompressor_model_outputs(temp_dir, save_compressed): def _check_llmcompressor_model_outputs(temp_dir, save_compressed):
# recipe.yaml should exist
assert (Path(temp_dir) / "recipe.yaml").exists()
# sparsity config exists if save_compressed
if save_compressed: if save_compressed:
assert (Path(temp_dir) / "recipe.yaml").exists()
from compressed_tensors import ModelCompressor from compressed_tensors import ModelCompressor
from compressed_tensors.config import Sparse24BitMaskConfig from compressed_tensors.config import Sparse24BitMaskConfig