Skip to content

Commit c649d35

Browse files
committed
clean
Signed-off-by: junq <[email protected]>
1 parent 293854a commit c649d35

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

tensorrt_llm/executor/base_worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def _create_py_executor():
128128
create_autodeploy_executor
129129
create_executor = create_autodeploy_executor
130130
assert isinstance(self.llm_args, ADLlmArgs)
131-
args["ad_config"] = self.llm_args.get_pytorch_backend_config()
131+
args["ad_config"] = self.llm_args
132132
args["tokenizer"] = self._tokenizer
133133
else:
134134
raise ValueError(f"Unsupported backend config: {self._backend}")

tests/unittest/_torch/auto_deploy/unit/singlegpu/shim/test_llm_config.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -65,12 +65,6 @@ def get_transform_config(free_mem_ratio):
6565
InferenceOptimizer(None, get_transform_config(1.1))
6666

6767

68-
def test_get_pytorch_backend_config():
69-
"""Test that get_pytorch_backend_config returns self."""
70-
args = LlmArgs(model="test-model")
71-
assert args.get_pytorch_backend_config() == args
72-
73-
7468
# ================================
7569
# Config Flow Tests
7670
# ================================

0 commit comments

Comments
 (0)