Skip to content

Commit d95478f

Browse files
committed
fix: Correct logic for checkpoint_format handling in create_input_processor
- Updated the condition to check for "HF" explicitly, removing the None case handling. Signed-off-by: Robin Kobus <[email protected]>
1 parent 6b5f526 commit d95478f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorrt_llm/inputs/registry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -419,7 +419,7 @@ def create_input_processor(
419419
model_path_or_dir: Path or repo id used to locate pretrained config/tokenizer.
420420
tokenizer: Tokenizer instance.
421421
checkpoint_format: Checkpoint format identifier. "HF" uses Hugging Face-style
422-
config loading; any other value skips HF config loading. None is treated as "HF".
422+
config loading; any other value skips HF config loading.
423423
424424
Returns:
425425
An InputProcessor implementation (model-specific if registered; otherwise DefaultInputProcessor).
@@ -429,7 +429,7 @@ def create_input_processor(
429429

430430
model_config = None
431431

432-
if checkpoint_format is None or checkpoint_format == "HF":
432+
if checkpoint_format == "HF":
433433
try:
434434
config = ModelConfig.from_pretrained(model_path_or_dir,
435435
trust_remote_code=True)

0 commit comments

Comments
 (0)