cumo/__init__.py

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