From be2e274ef504a09a622a4ea3bf7603b9a41b866f Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Tue, 11 Aug 2026 07:52:04 +0100 Subject: [PATCH] Fix docs on `main` (#51773) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> --- vllm/model_executor/models/transformers/base.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/vllm/model_executor/models/transformers/base.py b/vllm/model_executor/models/transformers/base.py index cb5d9dde59..1d0bbb71a6 100644 --- a/vllm/model_executor/models/transformers/base.py +++ b/vllm/model_executor/models/transformers/base.py @@ -279,14 +279,8 @@ class Base( )(cls) def _decorate_for_torch_compile(self): - """ - Decorate the model's decoder class to indicate to vLLM that it supports torch - compile if `can_enable_torch_compile` is True. - - Args: - kwargs: The kwargs to create the model, which are needed to get the decoder - class. - """ + """Decorate the model's decoder class to indicate to vLLM that it + supports torch compile if `can_enable_torch_compile` is True.""" self._decorate_cls_for_torch_compile( cls=self._pre_trained_model_classes.decoder, # Applied to a PreTrainedModel so the batch dimension will exist