clean up dreggs

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

src/llmcompressor/pipelines/basic/pipeline.py

@@ -10,10 +10,7 @@
 from llmcompressor.modifiers.utils.pytorch_helpers import apply_pad_mask_to_batch
 from llmcompressor.pipelines.registry import CalibrationPipeline
 from llmcompressor.pytorch.utils.helpers import tensors_to_device
-from llmcompressor.utils import (
-    calibration_forward_context,
-    dispatch_for_generation,
-)
+from llmcompressor.utils import calibration_forward_context, dispatch_for_generation
 
 if TYPE_CHECKING:
     from llmcompressor.args.dataset_arguments import DatasetArguments

src/llmcompressor/pipelines/sequential/pipeline.py

@@ -88,7 +88,6 @@ def __call__(
             # Optionally disable quantization
             if not dataset_args.quantization_aware_calibration or disable_qac:
                 stack.enter_context(DisableQuantization(model))
-            # Optionally disable lm_head
 
             # prepare intermediates cache
             activations = IntermediatesCache.from_dataloader(dataloader, model_device)