Skip to content

Commit d15b6a2

Browse files
committed
lint
Signed-off-by: Alex <[email protected]>
1 parent ced7516 commit d15b6a2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

vllm/model_executor/models/transformers.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ def init_buffers(self, module: nn.Module):
301301
setattr(module, name, new_buffer)
302302
for child in module.children():
303303
self.init_buffers(child)
304-
304+
305305
def init_parameters(self, module: nn.Module):
306306
"""
307307
If a `parameter` is on the `meta` device, then its parent
@@ -318,7 +318,9 @@ def init_parameters(self, module: nn.Module):
318318
"""
319319
for name, param in module.named_parameters(recurse=False):
320320
if param.device == torch.device("meta"):
321-
new_param = nn.Parameter(torch.empty_like(param.data, device=self.device_config.device))
321+
new_param = nn.Parameter(
322+
torch.empty_like(param.data,
323+
device=self.device_config.device))
322324
setattr(module, name, new_param)
323325
for child in module.children():
324326
self.init_parameters(child)

0 commit comments

Comments
 (0)