40 lines
1.3 KiB
Plaintext
40 lines
1.3 KiB
Plaintext
# API Reference {.doc .doc-index}
|
|
|
|
## Core
|
|
|
|
Core functionality for training
|
|
|
|
| | |
|
|
| --- | --- |
|
|
| [train](train.qmd#axolotl.train) | Prepare and train a model on a dataset. Can also infer from a model or merge lora |
|
|
| [evaluate](evaluate.qmd#axolotl.evaluate) | Module for evaluating models. |
|
|
| [datasets](datasets.qmd#axolotl.datasets) | Module containing Dataset functionality |
|
|
|
|
## CLI
|
|
|
|
Command-line interface
|
|
|
|
| | |
|
|
| --- | --- |
|
|
| [cli.main](cli.main.qmd#axolotl.cli.main) | Click CLI definitions for various axolotl commands. |
|
|
| [cli.train](cli.train.qmd#axolotl.cli.train) | CLI to run training on a model. |
|
|
| [cli.evaluate](cli.evaluate.qmd#axolotl.cli.evaluate) | CLI to run evaluation on a model. |
|
|
|
|
## Prompt Strategies
|
|
|
|
Prompt formatting strategies
|
|
|
|
| | |
|
|
| --- | --- |
|
|
| [prompt_strategies.base](prompt_strategies.base.qmd#axolotl.prompt_strategies.base) | module for base dataset transform strategies |
|
|
| [prompt_strategies.chat_template](prompt_strategies.chat_template.qmd#axolotl.prompt_strategies.chat_template) | HF Chat Templates prompt strategy |
|
|
|
|
## Utils
|
|
|
|
Utility functions
|
|
|
|
| | |
|
|
| --- | --- |
|
|
| [utils.models](utils.models.qmd#axolotl.utils.models) | Module for models and model loading |
|
|
| [utils.tokenization](utils.tokenization.qmd#axolotl.utils.tokenization) | Module for tokenization utilities |
|