Skip to content

Commit 94dced3

Browse files
justinchubyCopilot
andcommitted
Fix lint errors
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: Justin Chu <justinchu@microsoft.com>
1 parent c51e1cf commit 94dced3

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

src/mobius/models/gemma4.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1508,8 +1508,7 @@ def forward(
15081508
query_input = input_ids if input_ids is not None else hidden_states
15091509
fallback_bias_dict: dict[str, ir.Value | None] = {}
15101510
need_fallback = not use_gqa or any(
1511-
layer.self_attn.is_kv_shared_layer
1512-
for layer in self.layers
1511+
layer.self_attn.is_kv_shared_layer for layer in self.layers
15131512
)
15141513
if need_fallback:
15151514
if use_gqa:

0 commit comments

Comments
 (0)