Skip to content

Commit 6acb319

Browse files
committed
Update config.py
Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com>
1 parent a5993fc commit 6acb319

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

vllm/transformers_utils/config.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -691,11 +691,10 @@ def recurse_elems(elem: Any):
691691
max_position_embeddings = 128_000
692692
try:
693693
trust_remote_code_val = kwargs.get("trust_remote_code", False)
694-
hf_config = get_config(
695-
model=model,
696-
trust_remote_code=trust_remote_code_val,
697-
revision=revision,
698-
config_format=ConfigFormat.HF)
694+
hf_config = get_config(model=model,
695+
trust_remote_code=trust_remote_code_val,
696+
revision=revision,
697+
config_format=ConfigFormat.HF)
699698
if hf_value := hf_config.get_text_config().max_position_embeddings:
700699
max_position_embeddings = hf_value
701700
except Exception as e:

0 commit comments

Comments
 (0)