llava/__init__.py

from .model import LlavaLlamaForCausalLM, LlavaMistralForCausalLM
try:
    from .model import LlavaGemmaForCausalLM
except:
    pass
try:
    from .model import LlavaThothForCausalLM
except:
    pass