chouss
/

llava-spat

Model card Files Files and versions Community

llava-spat / llava /train /train_mem.py

chouss's picture

Uploading folder contents

754ac61 verified about 2 months ago

history blame contribute delete

115 Bytes

	from llava.train.train import train

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