Fix yapf detected format issue

Signed-off-by: Chendi.Xue <chendi.xue@intel.com>
red-hat-data-services · Sep 12, 2024 · 046cb25 · 046cb25
1 parent fb98cad
commit 046cb25
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/vllm/model_executor/models/dbrx.py b/vllm/model_executor/models/dbrx.py
@@ -82,17 +82,15 @@ def __init__(
 
         self.router = DbrxRouter(config, self.params_dtype)
         self.ws = nn.Parameter(
-            torch.empty(
-                self.num_total_experts,
-                2 * self.intermediate_size,
-                self.d_model,
-                dtype=self.params_dtype))
+            torch.empty(self.num_total_experts,
+                        2 * self.intermediate_size,
+                        self.d_model,
+                        dtype=self.params_dtype))
         self.w2s = nn.Parameter(
-            torch.empty(
-                self.num_total_experts,
-                self.d_model,
-                self.intermediate_size,
-                dtype=self.params_dtype))
+            torch.empty(self.num_total_experts,
+                        self.d_model,
+                        self.intermediate_size,
+                        dtype=self.params_dtype))
 
         set_weight_attrs(
             self.ws,