MQT-LLaVA / llava /train /train_mem.py
gordonhubackup's picture
init
140efa3
raw
history blame
115 Bytes
from llava.train.train import train
if __name__ == "__main__":
train(attn_implementation="flash_attention_2")