Spaces:

neuralcomputation
/

batik

Sleeping

batik / llava /train /train_mem.py

add llava to space

b8c85bc verified 4 months ago

119 Bytes

	from llava.train.train import train

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