Skip to content

Commit 60bab8a

Browse files
committed
refactor(huggingface): missing type hint
1 parent 9707027 commit 60bab8a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/modalities/conversion/gpt2/conversion_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def _get_layer_norm_value(config: dict, field: str) -> bool | float | int:
117117
return config.get(field, default)
118118

119119

120-
def _map_attention_type(config):
120+
def _map_attention_type(config: dict):
121121
if config["attention_implementation"] == "pytorch_flash":
122122
attention_impl = "sdpa"
123123
elif config["attention_implementation"] == "manual":

0 commit comments

Comments
 (0)