IntelLabs
/

LlavaOLMoBitnet1B

Model card Files Files and versions

LlavaOLMoBitnet1B / train /train_mem.py

naveensp's picture

Upload folder using huggingface_hub

d85b3f8 verified over 1 year ago

115 Bytes

	from llava.train.train import train

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