Skip to content

Commit f1ad7ea

Browse files
fix lint error
1 parent 7b986ed commit f1ad7ea

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

examples/models/gemma/convert_weights.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,7 @@
2626
}
2727

2828

29-
def gemma_to_executorch(
30-
state_dict: Dict[str, torch.Tensor]
31-
) -> Dict[str, torch.Tensor]:
29+
def gemma_to_executorch(state_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.Tensor]:
3230
"""
3331
Convert the state dict so that it matches what ExecuTorch's transformer definition expects.
3432
"""

0 commit comments

Comments
 (0)