llava/model/__init__.py

from .language_model.llava_llama import LlavaLlamaForCausalLM, LlavaConfig
from .language_model.llava_llamantk import LlavaLlamaNTKForCausalLM, LlavaNTKConfig

# from .language_model.llava_mpt import LlavaMPTForCausalLM, LlavaMPTConfig