Spaces:
Running
on
L4
Running
on
L4
| from llava.train.train import train | |
| if __name__ == "__main__": | |
| train(attn_implementation="flash_attention_2") | |