kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Remove unused import from kvcache.py

Browse Source 
Signed-off-by: Gal Hubara-Agam <96368689+galagam@users.noreply.github.com>
This commit is contained in:
Gal Hubara-Agam 2026-01-13 15:31:29 +02:00 committed by GitHub
parent 538f8cec1b
commit 5d77826300
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorrt_llm/_torch/auto_deploy/transform/library/kvcache.py
View File

@ -38,7 +38,7 @@ from ...models.factory import ModelFactory
from ...shim.interface import CachedSequenceInterface
from ...utils._graph import add_graph_input
from ...utils.cuda_mem_tracker import get_mem_info_in_mb
from ...utils.node_utils import get_all_input_output_nodes, is_op
from ...utils.node_utils import is_op
from ..interface import (
BaseTransform,
SharedConfig,