Skip to content

Commit 705da89

Browse files
author
Varun Sundar Rabindranath
committed
Remove redundant tp_size setting in dbrx
Signed-off-by: Varun Sundar Rabindranath <[email protected]>
1 parent 3b9c360 commit 705da89

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

vllm/model_executor/models/dbrx.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,6 @@ def __init__(
7979
prefix=prefix,
8080
)
8181
self.config = config
82-
self.tp_size = get_tensor_model_parallel_world_size()
8382
self.d_model = config.d_model
8483
self.intermediate_size = (self.config.ffn_config.ffn_hidden_size //
8584
self.tp_size)

0 commit comments

Comments
 (0)