Apply suggestions from code review

Co-authored-by: Huazhong Ji <hzji210@gmail.com>
2024-05-14 20:44:21 +08:00 · 2024-05-14 20:44:21 +08:00 · 9089bc70c8
parent 0ac6e73f99
commit 9089bc70c8
1 changed files with 1 additions and 1 deletions
--- a/src/train.py
+++ b/src/train.py
@ -1,5 +1,5 @@
 import os
-from torch_npu.contrib import transfer_to_npu
+from transformers import is_torch_npu_available
 from llmtuner.train.tuner import run_exp