add calib data for awq

kylesayrs · kylesayrs · commit 673fe047c7cf · 2025-05-05T16:00:28.000-04:00
Signed-off-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/src/llmcompressor/entrypoints/oneshot.py b/src/llmcompressor/entrypoints/oneshot.py
@@ -166,6 +166,7 @@ def apply_recipe_modifiers(
             recipe=self.recipe,
             recipe_stage=recipe_stage,
             recipe_args=self.recipe_args.recipe_args,
+            calib_data=calibration_dataloader,  # TODO: only used by AWQ modifier
         )
 
         user_pipeline = self.dataset_args.pipeline

Original file line number	Diff line number	Diff line change
`@@ -166,6 +166,7 @@ def apply_recipe_modifiers(`
`166`	`166`	`recipe=self.recipe,`
`167`	`167`	`recipe_stage=recipe_stage,`
`168`	`168`	`recipe_args=self.recipe_args.recipe_args,`
	`169`	`+ calib_data=calibration_dataloader, # TODO: only used by AWQ modifier`
`169`	`170`	`)`
`170`	`171`
`171`	`172`	`user_pipeline = self.dataset_args.pipeline`