VITA-1.5 / vita /model /__init__.py
lxysl's picture
upload vita-1.5 app.py
bc752b1
raw
history blame contribute delete
325 Bytes
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