> For the complete documentation index, see [llms.txt](https://axolotl.continuumlabs.pro/llms.txt). Markdown versions of documentation pages are available by appending `.md` to page URLs; this page is available as [Markdown](https://axolotl.continuumlabs.pro/model-selection-general.md).

# Model Selection - General

### <mark style="color:blue;">Model Configuration</mark>

The first configuration block of the <mark style="color:yellow;">Axolotl YAML configuration file</mark> is 'model type'.  It comprises three main configurations.

1. <mark style="color:blue;">**base\_model**</mark> -  This is the Huggingface model that contains \*.pt, \*.safetensors, or \*.bin files
2. <mark style="color:blue;">**model\_type**</mark> - If you want to specify the type of model to load
3. <mark style="color:blue;">**tokenizer\_type**</mark> - Corresponding tokenizer for the model&#x20;

<mark style="color:green;">Model Configuration example for Phi 2.0</mark>

```yaml
base_model: microsoft/phi-2
model_type: AutoModelForCausalLM
tokenizer_type: AutoTokenizer
```

<mark style="color:green;">Model Configuration example for Llama2</mark>

```yaml
base_model: NousResearch/Llama-2-7b-hf
model_type: LlamaForCausalLM
tokenizer_type: LlamaTokenizer
is_llama_derived_model: true
```

<mark style="color:green;">Model Configuration example for TinyLLama</mark>

```yaml
base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
model_type: LlamaForCausalLM
tokenizer_type: LlamaTokenizer
```

<mark style="color:green;">Model Configuration example for Code-LLama</mark>

```yaml
base_model: codellama/CodeLlama-7b-hf
model_type: LlamaForCausalLM
tokenizer_type: CodeLlamaTokenizer
```

We will now move to setting up the Axolotl training configuration file for Phi 2.0.


---

# Agent Instructions
This documentation is published with GitBook. GitBook is the documentation platform designed so that both humans and AI agents can read, navigate, and reason over technical content effectively. Learn more at gitbook.com.

## Querying This Documentation
If you need additional information that is not directly available in this page, you can query the documentation dynamically by asking a question.

Perform an HTTP GET request on the current page URL with the `ask` query parameter:

```
GET https://axolotl.continuumlabs.pro/model-selection-general.md?ask=<question>
```

The question should be specific, self-contained, and written in natural language.
The response will contain a direct answer to the question and relevant excerpts and sources from the documentation.

Use this mechanism when the answer is not explicitly present in the current page, you need clarification or additional context, or you want to retrieve related documentation sections.