GDinesh
/

llava-1-5

Inference Endpoints

Model card Files Files and versions Community

llava-1-5 / llava /train /train_mem.py

GDinesh's picture

Duplicate from saurabh-straive/llava-1-5

b2c8b1d verified 4 months ago

history blame contribute delete

No virus

115 Bytes

	from llava.train.train import train

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