Skip to content

Commit a751531

Browse files
committed
update broken link of PyTorchModelEngine in arch_overview
Signed-off-by: leslie-fang25 <[email protected]>
1 parent 2d2b8ba commit a751531

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/source/torch/arch_overview.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ The single-step flow of PyExecutor involves:
3737

3838
The core component of `PyExecutor` is the `ModelEngine`, responsible for executing the model's forward pass efficiently on the GPU.
3939
The key method of `ModelEngine` is `forward`, which handles the forward pass computation.
40-
For the PyTorch backend, the derived class is `PyTorchModelEngine`, declared in [pytorch_model_engine.py](../../../tensorrt_llm/_torch/pyexecutor/pytorch_model_engine.py).
40+
For the PyTorch backend, the derived class is `PyTorchModelEngine`, declared in [model_engine.py](../../../tensorrt_llm/_torch/pyexecutor/model_engine.py).
4141

4242
## Decoder
4343

0 commit comments

Comments
 (0)