Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com> Co-authored-by: Hongxia Yang <hongxia.yang@amd.com>
14 lines
232 B
Python
14 lines
232 B
Python
from vllm_omni.utils.platform_utils import (
|
|
detect_device_type,
|
|
get_device_control_env_var,
|
|
is_npu,
|
|
is_rocm,
|
|
)
|
|
|
|
__all__ = [
|
|
"detect_device_type",
|
|
"get_device_control_env_var",
|
|
"is_npu",
|
|
"is_rocm",
|
|
]
|