Formatting

HabanaAI · Sep 23, 2024 · a1c58e9 · a1c58e9
1 parent c80c18a
commit a1c58e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/worker/habana_model_runner.py b/vllm/worker/habana_model_runner.py
@@ -574,7 +574,8 @@ def _set_gc_threshold(self) -> None:
         self.multi_modal_input_mapper = MULTIMODAL_REGISTRY \
             .create_input_mapper(self.model_config)
 
-        self.skip_warmup = os.environ.get('VLLM_SKIP_WARMUP', 'false').lower() == 'true'
+        self.skip_warmup = os.environ.get('VLLM_SKIP_WARMUP',
+                                          'false').lower() == 'true'
 
     def load_model(self) -> None:
         import habana_frameworks.torch.core as htcore