merge main fix

vllm-project · Mar 5, 2025 · e5e292b · e5e292b
1 parent faeff7c
commit e5e292b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/llmcompressor/transformers/finetune/runner.py b/src/llmcompressor/transformers/finetune/runner.py
@@ -210,9 +210,9 @@ def run_sequential_stages(
 
                 calib_data = format_calibration_data(
                     tokenized_dataset=self.get_dataset_split("calibration"),
-                    num_calibration_samples=self._data_args.num_calibration_samples,
-                    do_shuffle=self._data_args.shuffle_calibration_samples,
-                    collate_fn=self._data_args.data_collator,
+                    num_calibration_samples=self._dataset_args.num_calibration_samples,
+                    do_shuffle=self._dataset_args.shuffle_calibration_samples,
+                    collate_fn=self._dataset_args.data_collator,
                 )
 
                 if do_preprocess: