cannot import name 'LlavaLlamaForCausalLM' from 'llava.model'
MKdir98 opened this issue · comments
Mahdi Karami commented
I'm getting this error in example code of llava-next
ImportError Traceback (most recent call last)
Cell In[5], line 1
----> 1 from llava.model.builder import load_pretrained_model
2 from llava.mm_utils import get_model_name_from_path, process_images, tokenizer_image_token
3 from llava.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN, IGNORE_INDEX
File /opt/conda/lib/python3.10/site-packages/llava/model/builder.py:23
21 import torch
22 from llava.model import *
---> 23 from llava.constants import DEFAULT_IMAGE_PATCH_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
24 from llava.utils import rank0_print
27 def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=False, device_map="auto", attn_implementation="flash_attention_2", customized_config=None, **kwargs):
File /opt/conda/lib/python3.10/site-packages/llava/__init__.py:1
----> 1 from .model import LlavaLlamaForCausalLM
ImportError: cannot import name 'LlavaLlamaForCausalLM' from 'llava.model' (/opt/conda/lib/python3.10/site-packages/llava/model/__init__.py)
Mahdi Karami commented
Kaichen Zhang - NTU commented
Is it something that related to this issue haotian-liu/LLaVA#1101?