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