Apply suggestions from code review

Co-authored-by: Huazhong Ji <hzji210@gmail.com>
Former-commit-id: 9089bc70c8838cb80473e557a750855f7b7a7695
This commit is contained in:
hoshi-hiyouga 2024-05-14 20:44:21 +08:00 committed by GitHub
parent 332f44fa43
commit fe586de344

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