MCPcopy Index your code
hub / github.com/XPixelGroup/DiffBIR / train_mem.py

File train_mem.py

llava/train/train_mem.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1from llava.train.train import train
2
3if __name__ == "__main__":
4 train(attn_implementation="flash_attention_2")

Callers

nothing calls this directly

Calls 1

trainFunction · 0.90

Tested by

no test coverage detected