Skip to content

Commit f23e871

Browse files
committed
reformat
1 parent 88e812a commit f23e871

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm/model_executor/models/modernbert.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,9 @@ def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
271271
elif self.pooling_type == "cls":
272272
pooled_output = pooled_output[0, :]
273273
else:
274-
raise ValueError(f"Pooling type should be either `cls` or `mean`, but got {self.pooling_type}")
274+
raise ValueError(
275+
f"Pooling type should be either `cls` or `mean`, but got {self.pooling_type}"
276+
)
275277
pooled_output = self.norm(self.act(self.dense(pooled_output)))
276278
return pooled_output
277279

0 commit comments

Comments
 (0)