File size: 119 Bytes
6351d80
 
 
 
1
2
3
4
5
from llava.train.train import train

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