Spaces:

gordonhu
/

MQT-LLaVA

MQT-LLaVA / llava /train /train_mem.py

init

140efa3 5 months ago

115 Bytes

	from llava.train.train import train

	if __name__ == "__main__":
	train(attn_implementation="flash_attention_2")