Spaces:

MBZUAI
/

LLaMA-3-V

Running on A10G

LLaMA-3-V / llava /train /train_mem.py

Adds code to host LLaVA-LLaMA-3 demo on HF space.

143d7a9 6 months ago

115 Bytes

	from llava.train.train import train

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