Spaces:
Running
on
Zero
Running
on
Zero
| from cumo.train.train import train | |
| if __name__ == "__main__": | |
| train(attn_implementation="flash_attention_2") | |