Skip to content

Commit 88ed188

Browse files
authored
typo
1 parent 173c924 commit 88ed188

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/exporters/openvino/model_patcher.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2501,7 +2501,7 @@ def __enter__(self):
25012501
_reinitialize_cos_sin_cached_fp32(layer.self_attn.rotary_emb)
25022502

25032503

2504-
# Adopted from https://github.com/huggingface/transformers/blob/31f9a289a6207be6cae746e009d8e0db523be203/src/transformers/models/falcon/modeling_falcon.py#L1138
2504+
# Adapted from https://github.com/huggingface/transformers/blob/31f9a289a6207be6cae746e009d8e0db523be203/src/transformers/models/falcon/modeling_falcon.py#L1138
25052505
def _falcon_prepare_4d_causal_attention_mask_with_cache_position(
25062506
attention_mask: torch.Tensor,
25072507
sequence_length: int,

0 commit comments

Comments
 (0)