Skip to content

llmcompressor.args.training_arguments

Training argument classes for LLM compression workflows.

This module defines dataclass-based argument containers for configuring training and one-shot calibration workflows. Extends HuggingFace's TrainingArguments with additional parameters specific to compression and stage-based execution.

Classes:

  • TrainingArguments

    Training arguments specific to LLM Compressor Transformers workflow using

TrainingArguments dataclass

TrainingArguments(
    do_oneshot: Optional[bool] = False,
    run_stages: Optional[bool] = False,
    output_dir: str = "./output",
)

Bases: TrainingArguments

Training arguments specific to LLM Compressor Transformers workflow using HFTrainingArgs as base class