from flash_vstream.model import VStreamLlamaForCausalLM