Format

Signed-off-by: Conroy Cheers <[email protected]>
vllm-project · Oct 29, 2024 · 4fcfec4 · 4fcfec4
1 parent c91ebd0
commit 4fcfec4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/vllm/platforms/__init__.py b/vllm/platforms/__init__.py
@@ -30,9 +30,11 @@
 except pynvml.NVMLError_LibraryNotFound:
     # CUDA is supported on Jetson, but NVML is not.
     import os
+
     def cuda_is_jetson() -> bool:
         return os.path.isfile("/etc/nv_tegra_release") \
             or os.path.exists("/sys/class/tegra-firmware")
+
     if cuda_is_jetson():
         is_cuda = True
         is_cuda_jetson = True

diff --git a/vllm/platforms/cuda_jetson.py b/vllm/platforms/cuda_jetson.py
@@ -6,13 +6,10 @@
 to fetch device information.
 """
 
-import os
 from functools import lru_cache
 
 import torch
 
-from vllm.logger import init_logger
-
 from .interface import DeviceCapability, Platform, PlatformEnum