Set statistics path to example directory in tiny_llama_find_hyperpara…

…ms (#3083) ### Changes Save statistics in example directory Add statistics path to .gitignore ### Tests https://github.com/openvinotoolkit/nncf/actions/runs/11822254518
openvinotoolkit · Nov 14, 2024 · 3d8b0e0 · 3d8b0e0
1 parent 95cf1ea
commit 3d8b0e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -129,6 +129,7 @@ examples/post_training_quantization/openvino/yolov8/yolov8n*
 examples/post_training_quantization/openvino/yolov8_quantize_with_accuracy_control/yolov8n*
 examples/**/runs/**
 examples/**/results/**
+examples/llm_compression/openvino/tiny_llama_find_hyperparams/statistics
 compressed_graph.dot
 original_graph.dot
 datasets/**

diff --git a/examples/llm_compression/openvino/tiny_llama_find_hyperparams/main.py b/examples/llm_compression/openvino/tiny_llama_find_hyperparams/main.py
@@ -31,6 +31,7 @@
 
 ROOT = Path(__file__).parent.resolve()
 MODEL_PATH = ROOT / "compressed_model.xml"
+STATISTICS_PATH = ROOT / "statistics"
 
 COMPRESSION_MODE = nncf.parameters.CompressWeightsMode.INT4_SYM
 MAX_DROP = 0.2
@@ -64,7 +65,7 @@ def compress_model(
         group_size=group_size,
         awq=awq,
         sensitivity_metric=nncf.parameters.SensitivityMetric.MAX_ACTIVATION_VARIANCE,
-        advanced_parameters=AdvancedCompressionParameters(statistics_path="statistics"),
+        advanced_parameters=AdvancedCompressionParameters(statistics_path=STATISTICS_PATH),
     )
     return optimized_ov_model