File size: 187 Bytes
9fa3d89
 
 
 
 
 
 
1
2
3
4
5
6
7
8
from ola_vlm.train.train import train

if __name__ == "__main__":
    # try:
        # train(attn_implementation="flash_attention_2")
    # except:
    train(attn_implementation="eager")