from .language_model.vita_mixtral import VITAMixtralConfig, VITAMixtralForCausalLM
from .language_model.vita_nemo import VITAMistralConfig, VITAMistralForCausalLM
from .language_model.vita_qwen2 import VITAQwen2Config, VITAQwen2ForCausalLM
from .language_model.vita_fo_qwen2 import VITAFOQwen2Config, VITAFOQwen2ForCausalLM

