We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 5d31244 + 5105475 commit dc5a243Copy full SHA for dc5a243
src/llmcompressor/entrypoints/oneshot.py
@@ -365,8 +365,8 @@ def _get_output_dir_from_argv() -> Optional[str]:
365
recipe_args.recipe_args = arg_dict
366
367
# raise depreciation warnings
368
- if dataset_args.remove_columns is not None:
369
- logger.waning(
+ if data_args.remove_columns is not None:
+ logger.warning(
370
"`remove_columns` argument is depreciated. When tokenizing datasets, all "
371
"columns which are invalid inputs the tokenizer will be removed",
372
DeprecationWarning,
0 commit comments