Spaces:

shi-labs
/

CuMo-7b-zero

Running on Zero

CuMo-7b-zero / cumo /train /train_mem.py

jiachenl

update

c3f3b0b 6 months ago

114 Bytes

	from cumo.train.train import train

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