Skip to content

Commit f2ff8c7

Browse files
committed
fix pre-commit
Signed-off-by: zt2370 <[email protected]>
1 parent 1d3a824 commit f2ff8c7

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm/model_executor/models/gpt2.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -315,12 +315,13 @@ def load_weights(self, weights: Iterable[tuple[str,
315315
loader = AutoWeightsLoader(self)
316316
weights = _add_transformer_prefix(weights)
317317
return loader.load_weights(weights)
318-
318+
319+
319320
def _add_transformer_prefix(
320321
weights: Iterable[tuple[str, torch.Tensor]]
321322
) -> Iterable[tuple[str, torch.Tensor]]:
322323
for name, tensor in weights:
323324
if not name.startswith('transformer.') and not name.startswith(
324325
"lm_head"):
325326
name = 'transformer.' + name
326-
yield name, tensor
327+
yield name, tensor

0 commit comments

Comments
 (0)