Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "5ed0cbe4ae9f56649a481857be208c0355b7841b"
Switch branch/tag
Find file Normal viewHistoryPermalink
train_mem.py 112 Bytes
Newer Older
dongchy920's avatar
MiniGemini_pytorch  
dongchy920 committed
1
2
3
4 
from mgm.train.train import train

if __name__ == "__main__":
    train(attn_implementation="flash_attention_2")