cli.args

cli.args

Module for axolotl CLI command arguments.

Classes

Name	Description
EvaluateCliArgs	Dataclass with CLI arguments for axolotl evaluate command.
InferenceCliArgs	Dataclass with CLI arguments for axolotl inference command.
PreprocessCliArgs	Dataclass with CLI arguments for axolotl preprocess command.
TrainerCliArgs	Dataclass with CLI arguments for axolotl train command.
VllmServeCliArgs	Dataclass with CLI arguments for axolotl vllm-serve command.

EvaluateCliArgs

cli.args.EvaluateCliArgs(
    self,
    debug=False,
    debug_text_only=False,
    debug_num_examples=0,
)

Dataclass with CLI arguments for axolotl evaluate command.

InferenceCliArgs

cli.args.InferenceCliArgs(self, prompter=None)

Dataclass with CLI arguments for axolotl inference command.

PreprocessCliArgs

cli.args.PreprocessCliArgs(
    self,
    debug=False,
    debug_text_only=False,
    debug_num_examples=1,
    prompter=None,
    download=True,
    iterable=None,
)

Dataclass with CLI arguments for axolotl preprocess command.

TrainerCliArgs

cli.args.TrainerCliArgs(
    self,
    debug=False,
    debug_text_only=False,
    debug_num_examples=0,
    merge_lora=False,
    prompter=None,
    shard=False,
    main_process_port=None,
    num_processes=None,
)

Dataclass with CLI arguments for axolotl train command.

VllmServeCliArgs

cli.args.VllmServeCliArgs(
    self,
    tensor_parallel_size=None,
    host=None,
    port=None,
    gpu_memory_utilization=None,
    dtype=None,
    max_model_len=None,
    enable_prefix_caching=None,
    serve_module=None,
)

Dataclass with CLI arguments for axolotl vllm-serve command.