train_mem.py

from blip3o.train.train import train

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