# Getting key-error:mistral when using autotrain

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgetting-key-error-mistral-when-using-autotrain%2F57881%3E%5C
**Category:** Models\
**Created:** [October 9, 2023, 8:56am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgetting-key-error-mistral-when-using-autotrain%2F57881 "2023-10-09T08:56:12Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![j1roscope](https://avatars.discourse-cdn.com/v4/letter/j/b9bd4f/32.png) [@j1roscope](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fj1roscope)%5C
**Post date:** [October 9, 2023, 8:56am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgetting-key-error-mistral-when-using-autotrain%2F57881%2F1 "2023-10-09T08:56:13Z")

</div>

when i try to fine tune a mistral model using autotrain using

```sh
 autotrain llm --train --project_name "medai_ft" --model TheBloke/Mistral-7B-OpenOrca-GGUF^C-data_path medalpaca/medical_meadow_medqa --text_column text --use_peft --use_int4 --learning_rate 2e-5 --train_batch_size 4 --num_train 5 --model_max_length 4096

```

i get an error

```sh
> ERROR train has failed due to an exception:
> ERROR Traceback (most recent call last):
  File "/home/divanshu/localGPT/venv/lib/python3.11/site-packages/autotrain/utils.py", line 280, in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
  File "/home/divanshu/localGPT/venv/lib/python3.11/site-packages/autotrain/trainers/clm/ __main__.py", line 78, in train
    tokenizer = AutoTokenizer.from_pretrained(
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/divanshu/localGPT/venv/lib/python3.11/site-packages/transformers/models/auto/tokenization_auto.py", line 701, in from_pretrained
    config = AutoConfig.from_pretrained(
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/divanshu/localGPT/venv/lib/python3.11/site-packages/transformers/models/auto/configuration_auto.py", line 1039, in from_pretrained
    config_class = CONFIG_MAPPING[config_dict["model_type"]]
                   ~~~~~~~~~~~~~~ ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/divanshu/localGPT/venv/lib/python3.11/site-packages/transformers/models/auto/configuration_auto.py", line 734, in __getitem__
    raise KeyError(key)
KeyError: 'mistral'

```

can anyone help me

---

<div class="post-metadata">

**Author:** ![Qisheng](https://avatars.discourse-cdn.com/v4/letter/q/e274bd/32.png) [@Qisheng](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FQisheng)%5C
**Post date:** [October 12, 2023, 6:39am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fgetting-key-error-mistral-when-using-autotrain%2F57881%2F2 "2023-10-12T06:39:05Z")

</div>

I also met this issue. Solved by update my transformers package to the latest version.