Skip to content

Commit 0b1af86

Browse files
committed
fix for dev transformers version
1 parent ddd35be commit 0b1af86

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
@@ -454,7 +454,7 @@ def _llama_gemma_update_causal_mask_latest(
454454

455455

456456
# TODO : deprecate _llama_gemma_update_causal_mask_legacy when transformers>=4.41.0
457-
if is_transformers_version(">=", "4.41.0"):
457+
if is_transformers_version(">", "4.40.0"):
458458
_llama_gemma_update_causal_mask = _llama_gemma_update_causal_mask_latest
459459
else:
460460
_llama_gemma_update_causal_mask = _llama_gemma_update_causal_mask_legacy

0 commit comments

Comments
 (0)