How to load the model
#3
by
lanfz2000
- opened
I used the code:
from transformers import AutoProcessor, AutoModelForCausalLM
processor = AutoProcessor.from_pretrained("jamessyx/PathGen-LLaVA")
model = AutoModelForCausalLM.from_pretrained("jamessyx/PathGen-LLaVA")
But it does not work.
Traceback (most recent call last):
File "/home/ubuntu/anaconda3/envs/llama/lib/python3.11/site-packages/transformers/models/auto/configuration_auto.py", line 1271, in from_pretrained
config_class = CONFIG_MAPPING[config_dict["model_type"]]
~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/ubuntu/anaconda3/envs/llama/lib/python3.11/site-packages/transformers/models/auto/configuration_auto.py", line 966, in getitem
raise KeyError(key)
KeyError: 'llava_llama'