format.sh

HabanaAI · Aug 13, 2024 · 08c9cf3 · 08c9cf3
1 parent db4c30f
commit 08c9cf3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/vllm/distributed/device_communicators/hpu_communicator.py b/vllm/distributed/device_communicators/hpu_communicator.py
@@ -5,9 +5,8 @@
 from vllm.platforms import current_platform
 from vllm.utils import is_fake_hpu
 
-if current_platform.is_hpu():
-    if not is_fake_hpu():
-        import habana_frameworks.torch as htorch  # noqa: F401
+if current_platform.is_hpu() and not is_fake_hpu():
+    import habana_frameworks.torch as htorch  # noqa: F401
 
 
 class HpuCommunicator:

diff --git a/vllm/executor/ray_habana_executor.py b/vllm/executor/ray_habana_executor.py
@@ -88,7 +88,7 @@ def _init_workers_ray(self, placement_group: "PlacementGroup",
         worker_wrapper_kwargs = self._get_worker_wrapper_args()
         for bundle_id, bundle in enumerate(placement_group.bundle_specs):
             resource_name = "HPU" if not is_fake_hpu() else "CPU"
-            if not bundle.get(resource_name,0):
+            if not bundle.get(resource_name, 0):
                 continue
             scheduling_strategy = PlacementGroupSchedulingStrategy(
                 placement_group=placement_group,