Files
llm_trainer/src/train.py
2024-05-03 02:58:23 +08:00

15 lines
171 B
Python

from llmtuner.train.tuner import run_exp
def main():
run_exp()
def _mp_fn(index):
# For xla_spawn (TPUs)
run_exp()
if __name__ == "__main__":
main()