issue of download your model

[yxu1168]

Hi Xin, Great job! I am trying to download your model to local drive to run app.py and chat.py, xinlai / LISA-13B-llama2-v1

But got KeyError: 'llava': KeyError Traceback (most recent call last) in <cell line: 4>() 2 3 tokenizer = AutoTokenizer.from_pretrained("xinlai/LISA-13B-llama2-v1") ----> 4 model = AutoModel.from_pretrained("xinlai/LISA-13B-llama2-v1") 5 6 # Save the model and tokenizer to your local drive

2 frames /usr/local/lib/python3.10/dist-packages/transformers/models/auto/configuration_auto.py in getitem(self, key) 708 return self._extra_content[key] 709 if key not in self._mapping: --> 710 raise KeyError(key) 711 value = self._mapping[key] 712 module_name = model_type_to_module_name(key)

KeyError: 'llava'

Do you know what happened?

The code I used: pip install transformers

from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("xinlai/LISA-13B-llama2-v1") model = AutoModel.from_pretrained("xinlai/LISA-13B-llama2-v1")

Thanks a lot.