Apply suggestions from code review

Co-authored-by: Huazhong Ji <hzji210@gmail.com>
Former-commit-id: abef48c17ee795eae984fcc89019c2c4859108c1
This commit is contained in:
hoshi-hiyouga 2024-05-14 20:44:21 +08:00 committed by GitHub
parent fe1a3b1367
commit 86a048128b

View File

@ -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