mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-06-22 07:08:57 +08:00
[misc] bump transformers version upperbound (#10446)
This commit is contained in:
@@ -20,6 +20,7 @@ import importlib.util
|
||||
from functools import lru_cache
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
import transformers.utils.import_utils as import_utils
|
||||
from packaging import version
|
||||
|
||||
|
||||
@@ -126,3 +127,26 @@ def is_uvicorn_available():
|
||||
|
||||
def is_vllm_available():
|
||||
return _is_package_available("vllm")
|
||||
|
||||
|
||||
_orig_is_package_available = import_utils._is_package_available
|
||||
|
||||
|
||||
class PackageAvailability(tuple):
|
||||
__slots__ = ()
|
||||
|
||||
def __new__(cls, available: bool, pkg_version: str = "N/A"):
|
||||
return super().__new__(cls, (bool(available), pkg_version))
|
||||
|
||||
def __bool__(self) -> bool:
|
||||
return self[0]
|
||||
|
||||
|
||||
def _patched_is_package_available(pkg_name: str, return_version: bool = False):
|
||||
available, version = _orig_is_package_available(pkg_name, return_version=return_version)
|
||||
|
||||
return PackageAvailability(available, version)
|
||||
|
||||
|
||||
if is_transformers_version_greater_than("5.3.0"):
|
||||
import_utils._is_package_available = _patched_is_package_available
|
||||
|
||||
Reference in New Issue
Block a user