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'

Sign up or log in to comment