LLaMA-3-V / llava /train /train_mem.py
mmaaz60's picture
Adds code to host LLaVA-LLaMA-3 demo on HF space.
143d7a9
from llava.train.train import train
if __name__ == "__main__":
train(attn_implementation="flash_attention_2")