File size: 115 Bytes
625a17f | 1 2 3 4 5 | from llava.train.train import train
if __name__ == "__main__":
train(attn_implementation="flash_attention_2")
|
625a17f | 1 2 3 4 5 | from llava.train.train import train
if __name__ == "__main__":
train(attn_implementation="flash_attention_2")
|