Skip to content

Commit

Permalink
Fix format
Browse files Browse the repository at this point in the history
Signed-off-by: Pooya Davoodi <[email protected]>
  • Loading branch information
pooyadavoodi committed Dec 9, 2024
1 parent e84c2a4 commit 7a06527
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions vllm/model_executor/models/gritlm.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,8 +212,7 @@ def __init__(
if hasattr(layer, "self_attn"):
assert isinstance(layer.self_attn.attn.impl, XFormersImpl), (
"GritLM is only supported by XFormers backend, "
"which can be forced by VLLM_ATTENTION_BACKEND=XFORMERS"
)
"which can be forced by VLLM_ATTENTION_BACKEND=XFORMERS")

def forward(
self,
Expand Down

0 comments on commit 7a06527

Please sign in to comment.