Skip to content

Commit a75b609

Browse files
eaidovaecharlaix
andauthored
Update optimum/exporters/openvino/model_patcher.py
Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
1 parent 9917b6b commit a75b609

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
@@ -951,7 +951,7 @@ def __exit__(self, exc_type, exc_value, traceback):
951951
block.attention.forward = block.attention._orig_forward
952952

953953

954-
# Adapted from Phi3Attention.forward
954+
# Adapted from https://github.com/huggingface/transformers/blob/ccdabc5642bf84849af93f591e207dc625c8e1e1/src/transformers/models/phi3/modeling_phi3.py#L426
955955
def _phi3_self_attn_sdpa_forward(
956956
self,
957957
hidden_states: torch.Tensor,

0 commit comments

Comments
 (0)