Skip to content

Commit 31cffb0

Browse files
committed
add args
1 parent f1ec15b commit 31cffb0

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/llmcompressor/args/model_arguments.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,4 +86,4 @@ class ModelArguments:
8686
"help": "The specific model version to use "
8787
"(can be a branch name, tag name or commit id)"
8888
},
89-
)
89+
)

src/llmcompressor/entrypoints/oneshot.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,8 @@ def apply_recipe_modifiers(
190190
user_pipeline = self.dataset_args.pipeline
191191
modifiers = session.get_modifiers()
192192
pipeline = CalibrationPipeline.from_modifiers(modifiers, user=user_pipeline)
193-
pipeline(self.model, calibration_dataloader, self.dataset_args)
193+
# ToDo: wrap moe_calibrate_all_experts in some set of args
194+
pipeline(self.model, calibration_dataloader, self.dataset_args, moe_calibrate_all_experts=True, moe_calibrate_gated_acts=True)
194195

195196
session.finalize()
196197

0 commit comments

Comments
 (0)