from xtuner.tools.train import main as train
try:
    import torch
    import torch_npu
    from torch_npu.contrib import transfer_to_npu
except:
    pass
if __name__ == '__main__':
    train()