Spaces:

wcy1122
/

MGM

Runtime error

MGM / minigemini /train /train_mem.py

update code

35153f6 3 months ago

No virus

119 Bytes

	from minigemini.train.train import train

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