Skip to content

Commit 6232596

Browse files
committed
Fix rename
Signed-off-by: Amit Zuker <[email protected]>
1 parent 6b9d9f3 commit 6232596

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tensorrt_llm/_torch/pyexecutor/_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,7 +482,7 @@ def create_py_executor_instance(
482482
num_lora_modules = model_engine.model.model_config.pretrained_config.num_hidden_layers * \
483483
len(lora_config.lora_target_modules + lora_config.missing_qkv_modules)
484484

485-
peft_cache_config_model = PeftCacheConfig.create_from_pybind(
485+
peft_cache_config_model = PeftCacheConfig.from_pybind(
486486
executor_config.peft_cache_config
487487
) if executor_config.peft_cache_config is not None else PeftCacheConfig(
488488
)

tensorrt_llm/llmapi/llm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -823,7 +823,7 @@ def _build_model(self):
823823
num_lora_modules = engine_config.pretrained_config.num_hidden_layers * \
824824
len(lora_config.lora_target_modules + lora_config.missing_qkv_modules)
825825

826-
peft_cache_config_model = PeftCacheConfig.create_from_pybind(
826+
peft_cache_config_model = PeftCacheConfig.from_pybind(
827827
self._executor_config.peft_cache_config
828828
) if self._executor_config.peft_cache_config is not None else PeftCacheConfig(
829829
)

0 commit comments

Comments
 (0)