Update optimum/exporters/openvino/model_patcher.py

eaidova · web-flow · commit 7428615fbbca · 2024-12-25T18:05:34.000+04:00
diff --git a/optimum/exporters/openvino/model_patcher.py b/optimum/exporters/openvino/model_patcher.py
@@ -2678,7 +2678,7 @@ def _gpt_neo_attn_forward(
     )
 
 
-# Adopted from https://github.com/huggingface/optimum/blob/main/optimum/bettertransformer/models/attention.py#L185
+# Adapted from https://github.com/huggingface/optimum/blob/main/optimum/bettertransformer/models/attention.py#L185
 def _gpt_neo_attn_sdpa(
     self,
     query: torch.Tensor,

Original file line number	Diff line number	Diff line change
`@@ -2678,7 +2678,7 @@ def _gpt_neo_attn_forward(`
`2678`	`2678`	`)`
`2679`	`2679`
`2680`	`2680`
`2681`		`-# Adopted from https://github.com/huggingface/optimum/blob/main/optimum/bettertransformer/models/attention.py#L185`
	`2681`	`+# Adapted from https://github.com/huggingface/optimum/blob/main/optimum/bettertransformer/models/attention.py#L185`
`2682`	`2682`	`def _gpt_neo_attn_sdpa(`
`2683`	`2683`	`self,`
`2684`	`2684`	`query: torch.Tensor,`