From fe586de3446ffbcc00b28ed654f069e3132780d5 Mon Sep 17 00:00:00 2001
From: hoshi-hiyouga <hiyouga@buaa.edu.cn>
Date: Tue, 14 May 2024 20:44:21 +0800
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: Huazhong Ji <hzji210@gmail.com>
Former-commit-id: 9089bc70c8838cb80473e557a750855f7b7a7695
---
 src/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/train.py b/src/train.py
index 098ec1b5..00a7fa26 100644
--- 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