From 8eb243cc38bd3a3cfbe7a294c9954e532f169311 Mon Sep 17 00:00:00 2001 From: Thomas Parnell Date: Sat, 3 Aug 2024 02:32:21 -0400 Subject: [PATCH] fmt Signed-off-by: Thomas Parnell --- vllm/model_executor/models/mlp_speculator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vllm/model_executor/models/mlp_speculator.py b/vllm/model_executor/models/mlp_speculator.py index b8bed2d72371b..95a655fbbf37d 100644 --- a/vllm/model_executor/models/mlp_speculator.py +++ b/vllm/model_executor/models/mlp_speculator.py @@ -65,6 +65,7 @@ class MLPSpeculator(nn.Module): Trained speculators of this type are available on HF hub at: https://huggingface.co/ibm-fms and https://huggingface.co/ibm-granite """ + def __init__(self, config: MLPSpeculatorConfig, **kwargs) -> None: super().__init__() self.n_predict = config.n_predict