format

Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>
ROCm · Feb 1, 2025 · 5fe1d1d · 5fe1d1d
1 parent 0d66687
commit 5fe1d1d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/vllm/model_executor/model_loader/loader.py b/vllm/model_executor/model_loader/loader.py
@@ -1386,8 +1386,7 @@ def load_model(self, vllm_config: VllmConfig) -> nn.Module:
                     hasattr(module, "process_weights_after_loading"):
                     # When attention modules need to process weights after
                     # currently only used by MLA
-                    module.process_weights_after_loading(
-                            model_config.dtype)
+                    module.process_weights_after_loading(model_config.dtype)
         return model.eval()