Lint again, to fixed point

ROCm · Dec 19, 2024 · 06f53ba · 06f53ba
1 parent 1ed1389
commit 06f53ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/model_executor/models/llama.py b/vllm/model_executor/models/llama.py
@@ -231,8 +231,8 @@ def forward(
                                 attn_metadata,
                                 fp8_comp_scales=(self.attn._q_scale,
                                                  self.attn._prob_scale,
-                                                 self.o_proj.input_scale
-                                                 if self.attn_fp8_out else None))
+                                                 self.o_proj.input_scale if
+                                                 self.attn_fp8_out else None))
         output, _ = self.o_proj(attn_output)
         return output