Spaces:
Running
on
Zero
Running
on
Zero
File size: 114 Bytes
d1f015b |
1 2 3 4 5 |
from cumo.train.train import train
if __name__ == "__main__":
train(attn_implementation="flash_attention_2")
|