423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-16 00:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

The training efficiency of the Ascend 910A has been significantly enhanced, leveraging the full computational power of the NPU (Neural Processing Unit) and the capabilities of torch_npu, a PyTorch library optimized for NPUs. This improvement has resulted in a remarkable tenfold increase in efficiency.

Browse Source 
Former-commit-id: 90980b626d3408b3e2ee32a02456c20881318be7
This commit is contained in:
zhouwei 2024-05-06 13:29:59 +08:00
parent 25de4ce56a
commit 3d1b0e1864
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/train.py
View File

@ -1,3 +1,7 @@
import os
import torch
import torch_npu
from torch_npu.contrib import transfer_to_npu
from llmtuner.train.tuner import run_exp
@ -11,4 +15,6 @@ def _mp_fn(index):
if __name__ == "__main__":
use_jit_compile = os.getenv('JIT_COMPILE', 'False').lower() in ['true', '1']
torch.npu.set_compile_mode(jit_compile=use_jit_compile)
main()