bugfix

Signed-off-by: Brian Dellabetta <[email protected]>

Author: brian-dellabetta

src/llmcompressor/entrypoints/oneshot.py

@@ -125,7 +125,7 @@ def __init__(
         self.output_dir = output_dir
 
         # initialize the model and processor
-        pre_process(model_args)
+        pre_process(model_args, dataset_args)
 
         # Set instance attributes
         self.model = self.model_args.model

src/llmcompressor/entrypoints/utils.py

@@ -16,10 +16,10 @@
 from transformers.utils.quantization_config import CompressedTensorsConfig
 
 from llmcompressor.args import (
+    DatasetArguments,
     ModelArguments,
     RecipeArguments,
     TrainingArguments,
-    DatasetArguments,
 )
 from llmcompressor.core import reset_session
 from llmcompressor.pytorch.model_load.helpers import parse_dtype