9fa3d89
1
2
3
4
5
6
7
8
from ola_vlm.train.probe_train import train if __name__ == "__main__": try: train(attn_implementation="flash_attention_2") except: train(attn_implementation="eager")
