Spaces:

Senqiao
/

VisionZip

Sleeping

VisionZip / llava /train /train_mem.py

Release code same to the github

025f5af 6 months ago

115 Bytes

	from llava.train.train import train

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