> 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.