Pass template argument

Signed-off-by: Thomas Parnell <[email protected]>
IBM · Jan 21, 2025 · 9703645 · 9703645
1 parent 6e8de27
commit 9703645
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/worker/spyre_model_runner.py b/vllm/worker/spyre_model_runner.py
@@ -64,7 +64,7 @@ def from_broadcasted_tensor_dict(
         return cls(**tensor_dict)
 
 
-class SpyreModelRunner(ModelRunnerBase):
+class SpyreModelRunner(ModelRunnerBase[ModelInputForSpyre]):
 
     def __init__(
         self,