Fix docs on `main` (#51773)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Harry Mellor committed
Aug 11, 2026 at 07:52 UTC
be2e274ef504a09a622a4ea3bf7603b9a41b866f
1 file changed
+2
-8
vllm/model_executor/models/transformers/base.py
+2
-8
@@ -279,14 +279,8 @@ class Base(
279
)(cls)
280
281
def _decorate_for_torch_compile(self):
282
- """
283
- Decorate the model's decoder class to indicate to vLLM that it supports torch
284
- compile if `can_enable_torch_compile` is True.
285
-
286
- Args:
287
- kwargs: The kwargs to create the model, which are needed to get the decoder
288
- class.
289
- """
282
+ """Decorate the model's decoder class to indicate to vLLM that it
283
+ supports torch compile if `can_enable_torch_compile` is True."""
284
self._decorate_cls_for_torch_compile(
285
cls=self._pre_trained_model_classes.decoder,
286
# Applied to a PreTrainedModel so the batch dimension will exist