Formatting

vllm-project · Jul 16, 2024 · 080d588 · 080d588
1 parent 2de01e3
commit 080d588
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/model_executor/models/llama.py b/vllm/model_executor/models/llama.py
@@ -385,7 +385,7 @@ def __init__(
 
             logit_scale = getattr(config, "logit_scale", 1.0)
             self.logits_processor = LogitsProcessor(self.unpadded_vocab_size,
-                                                    config.vocab_size, logit_scale)
+                                                config.vocab_size, logit_scale)
             self.sampler = Sampler()
         else:
             self.lm_head = PPMissingLayer()