from diffusion.model.llava.llava_mpt import LlavaMPTForCausalLM, LlavaMPTConfig