comments

George Ohashi · George Ohashi · commit ab8aacfe8d76 · 2025-03-05T11:52:02.000-05:00
diff --git a/src/llmcompressor/transformers/finetune/runner.py b/src/llmcompressor/transformers/finetune/runner.py
@@ -213,6 +213,9 @@ def run_sequential_stages(
                     num_calibration_samples=self._dataset_args.num_calibration_samples,
                     do_shuffle=self._dataset_args.shuffle_calibration_samples,
                     collate_fn=self._dataset_args.data_collator,
+                    accelerator=self.trainer.accelerator
+                    if self.trainer is not None
+                    else None,
                 )
 
                 if do_preprocess:

Original file line number	Diff line number	Diff line change
`@@ -213,6 +213,9 @@ def run_sequential_stages(`
`213`	`213`	`num_calibration_samples=self._dataset_args.num_calibration_samples,`
`214`	`214`	`do_shuffle=self._dataset_args.shuffle_calibration_samples,`
`215`	`215`	`collate_fn=self._dataset_args.data_collator,`
	`216`	`+ accelerator=self.trainer.accelerator`
	`217`	`+ if self.trainer is not None`
	`218`	`+ else None,`
`216`	`219`	`)`
`217`	`220`
`218`	`221`	`if do_preprocess:`