Spaces:

Yiyuan
/

VSA

Sleeping

VSA / models /LLaVA /llava /train /train_mem.py

init

b27b0a2 9 days ago

115 Bytes

	from llava.train.train import train

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