Skip to content

Commit 5b4dff7

Browse files
committed
actually fix parameter initialization
Signed-off-by: Alex <[email protected]>
1 parent 1ea0238 commit 5b4dff7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/models/transformers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ def init_parameters(self, module: nn.Module):
317317
"""
318318
for name, param in module.named_parameters(recurse=False):
319319
if param.device == torch.device("meta"):
320-
new_param = getattr(type(module)(self.config), name)
320+
new_param = nn.Parameter(torch.empty_like(param.data, device=self.device_config.device))
321321
setattr(module, name, new_param)
322322
for child in module.children():
323323
self.init_parameters(child)

0 commit comments

Comments
 (0)