Skip to content

Commit 43a9074

Browse files
committed
lint
Signed-off-by: Alex <[email protected]>
1 parent 5b4dff7 commit 43a9074

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
@@ -300,7 +300,7 @@ def init_buffers(self, module: nn.Module):
300300
setattr(module, name, new_buffer)
301301
for child in module.children():
302302
self.init_buffers(child)
303-
303+
304304
def init_parameters(self, module: nn.Module):
305305
"""
306306
If a `parameter` is on the `meta` device, then its parent
@@ -317,7 +317,9 @@ 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 = nn.Parameter(torch.empty_like(param.data, device=self.device_config.device))
320+
new_param = nn.Parameter(
321+
torch.empty_like(param.data,
322+
device=self.device_config.device))
321323
setattr(module, name, new_param)
322324
for child in module.children():
323325
self.init_parameters(child)

0 commit comments

Comments
 (0)