mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-12-29 10:10:35 +08:00
[model] update kt code (#9406)
This commit is contained in:
@@ -312,9 +312,11 @@ def use_openmind() -> bool:
|
||||
def use_ray() -> bool:
|
||||
return is_env_enabled("USE_RAY")
|
||||
|
||||
|
||||
def use_kt() -> bool:
|
||||
return is_env_enabled("USE_KT")
|
||||
|
||||
|
||||
def find_available_port() -> int:
|
||||
r"""Find an available port on the local machine."""
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
|
||||
@@ -110,6 +110,7 @@ def is_starlette_available():
|
||||
def is_transformers_version_greater_than(content: str):
|
||||
return _get_package_version("transformers") >= version.parse(content)
|
||||
|
||||
|
||||
@lru_cache
|
||||
def is_torch_version_greater_than(content: str):
|
||||
return _get_package_version("torch") >= version.parse(content)
|
||||
|
||||
Reference in New Issue
Block a user