lint

Former-commit-id: 9030501eaef97ea249347198272adf0d709503ec

src/llamafactory/extras/env.py

@@ -6,9 +6,10 @@ import peft
 import torch
 import transformers
 import trl
+from transformers.integrations import is_deepspeed_available
 from transformers.utils import is_bitsandbytes_available, is_torch_cuda_available, is_torch_npu_available
 
-from .packages import is_deepspeed_available, is_vllm_available
+from .packages import is_vllm_available
 
 
 VERSION = "0.7.2.dev0"

src/llamafactory/extras/packages.py

@@ -20,10 +20,6 @@ def _get_package_version(name: str) -> "Version":
         return version.parse("0.0.0")
 
 
-def is_deepspeed_available():
-    return _is_package_available("deepspeed")
-
-
 def is_fastapi_available():
     return _is_package_available("fastapi")