Duplicated from merve/llava-next
6a83074
1
2
3
4
5
from llava_llama3.train.train import train if __name__ == "__main__": train(attn_implementation="flash_attention_2")
