T

Diffusers
Safetensors
T / llava /train /train_mem.py
chenzeyang1's picture
Add files using upload-large-folder tool
964f845 verified
Raw
History Blame Contribute Delete
115 Bytes
from llava.train.train import train
if __name__ == "__main__":
train(attn_implementation="flash_attention_2")