Skip to content

Commit 52bda05

Browse files
committed
fix param dtype
Signed-off-by: Isotr0py <2037008807@qq.com>
1 parent cfa1998 commit 52bda05

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm/model_executor/models/transformers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -599,6 +599,7 @@ def init_parameters(self, module: nn.Module):
599599
if param.device == torch.device("meta"):
600600
new_param = nn.Parameter(
601601
torch.empty_like(param.data,
602+
dtype=self.model_config.dtype,
602603
device=self.device_config.device))
603604
setattr(module, name, new_param)
604605
for child in module.children():

0 commit comments

Comments
 (0)