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. |
QuantizeCliArgs | Dataclass with CLI arguments for axolotl quantize command. |
TrainerCliArgs | Dataclass with CLI arguments for axolotl train command. |
VllmServeCliArgs | Dataclass with CLI arguments for axolotl vllm-serve command. |
EvaluateCliArgs
cli.args.EvaluateCliArgs(=False,
debug=False,
debug_text_only=0,
debug_num_examples )
Dataclass with CLI arguments for axolotl evaluate
command.
InferenceCliArgs
=None) cli.args.InferenceCliArgs(prompter
Dataclass with CLI arguments for axolotl inference
command.
PreprocessCliArgs
cli.args.PreprocessCliArgs(=False,
debug=False,
debug_text_only=1,
debug_num_examples=None,
prompter=True,
download=None,
iterable )
Dataclass with CLI arguments for axolotl preprocess
command.
QuantizeCliArgs
cli.args.QuantizeCliArgs(=None,
base_model=None,
weight_dtype=None,
activation_dtype=None,
quantize_embedding=None,
group_size=None,
output_dir )
Dataclass with CLI arguments for axolotl quantize
command.
TrainerCliArgs
cli.args.TrainerCliArgs(=False,
debug=False,
debug_text_only=0,
debug_num_examples=False,
merge_lora=None,
prompter=False,
shard=None,
main_process_port=None,
num_processes )
Dataclass with CLI arguments for axolotl train
command.
VllmServeCliArgs
cli.args.VllmServeCliArgs(=None,
tensor_parallel_size=None,
host=None,
port=None,
gpu_memory_utilization=None,
dtype=None,
max_model_len=None,
enable_prefix_caching=None,
serve_module )
Dataclass with CLI arguments for axolotl vllm-serve
command.