diff --git a/openllm-python/src/openllm/utils.py b/openllm-python/src/openllm/utils.py index 7dda7858..f23b26b4 100644 --- a/openllm-python/src/openllm/utils.py +++ b/openllm-python/src/openllm/utils.py @@ -16,7 +16,7 @@ def generate_labels(llm): def available_devices(): - from .._strategies import NvidiaGpuResource + from ._strategies import NvidiaGpuResource return tuple(NvidiaGpuResource.from_system())