| from omni_speech.train.train_multiturn import train | |
| if __name__ == "__main__": | |
| train(attn_implementation="flash_attention_2") |
| from omni_speech.train.train_multiturn import train | |
| if __name__ == "__main__": | |
| train(attn_implementation="flash_attention_2") |