Can anyone show me how to download this? I tried using the cache_dir="./mydir/" but it seems so slow when downloading.

example of the code I use:

Load model directly

from transformers import AutoModelForCausalLM
model_path = "/home/bdina/Desktop/GEO-ChatGPT/lava_weights"
model = AutoModelForCausalLM.from_pretrained(model_path)

Why do I get this error:

Exception has occurred: KeyError (note: full exception trace is shown but execution is paused at: _run_module_as_main)
'llava'
File "/home/bdina/anaconda3/envs/video_chatgpt/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 748, in getitem
raise KeyError(key)
File "/home/bdina/anaconda3/envs/video_chatgpt/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 1050, in from_pretrained
config_class = CONFIG_MAPPING[config_dict["model_type"]]
File "/home/bdina/anaconda3/envs/video_chatgpt/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 525, in from_pretrained
config, kwargs = AutoConfig.from_pretrained(
File "/home/bdina/Desktop/GEO-ChatGPT/data/llava_test.py", line 5, in
model = AutoModelForCausalLM.from_pretrained(model_path)
File "/home/bdina/anaconda3/envs/video_chatgpt/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
File "/home/USER/anaconda3/envs/video_chatgpt/lib/python3.10/runpy.py", line 196, in _run_module_as_main (Current frame)
return _run_code(code, main_globals, None,
KeyError: 'llava'