Remove redundant tp_size setting in dbrx

Varun Sundar Rabindranath · Varun Sundar Rabindranath · commit 705da89bd4b9 · 2025-05-14T03:29:02.000-04:00
Signed-off-by: Varun Sundar Rabindranath &lt;varun@neuralmagic.com&gt;
diff --git a/vllm/model_executor/models/dbrx.py b/vllm/model_executor/models/dbrx.py
@@ -79,7 +79,6 @@ def __init__(
             prefix=prefix,
         )
         self.config = config
-        self.tp_size = get_tensor_model_parallel_world_size()
         self.d_model = config.d_model
         self.intermediate_size = (self.config.ffn_config.ffn_hidden_size //
                                   self.tp_size)

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,6 @@ def __init__(`
`79`	`79`	`prefix=prefix,`
`80`	`80`	`)`
`81`	`81`	`self.config = config`
`82`		`- self.tp_size = get_tensor_model_parallel_world_size()`
`83`	`82`	`self.d_model = config.d_model`
`84`	`83`	`self.intermediate_size = (self.config.ffn_config.ffn_hidden_size //`
`85`	`84`	`self.tp_size)`