Skip to content

Commit 658d463

Browse files
committed
fix format.
Signed-off-by: Fanrong Li <[email protected]>
1 parent 6a638aa commit 658d463

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tensorrt_llm/_torch/model_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -426,8 +426,10 @@ def from_pretrained(cls,
426426
trust_remote_code=trust_remote_code,
427427
**kwargs,
428428
)
429-
if pretrained_config.architectures[0] == "DeepseekV32ForCausalLM":
430-
sparse_attention_config = kwargs.get('sparse_attention_config')
429+
if pretrained_config.architectures[
430+
0] == "DeepseekV32ForCausalLM":
431+
sparse_attention_config = kwargs.get(
432+
'sparse_attention_config')
431433
if sparse_attention_config:
432434
index_n_heads = sparse_attention_config.index_n_heads or pretrained_config.index_n_heads
433435
index_head_dim = sparse_attention_config.index_head_dim or pretrained_config.index_head_dim

0 commit comments

Comments
 (0)