Spaces:
Runtime error
Runtime error
| from cumo.train.train import train | |
| if __name__ == "__main__": | |
| train(attn_implementation="flash_attention_2") | |
| from cumo.train.train import train | |
| if __name__ == "__main__": | |
| train(attn_implementation="flash_attention_2") | |