FinLLaVA / llava_llama3 /train /train_mem.py
TobyYang7's picture
Upload 96 files
6a83074 verified
raw
history blame
122 Bytes
from llava_llama3.train.train import train
if __name__ == "__main__":
train(attn_implementation="flash_attention_2")