We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ddd35be commit 0b1af86Copy full SHA for 0b1af86
optimum/exporters/openvino/model_patcher.py
@@ -454,7 +454,7 @@ def _llama_gemma_update_causal_mask_latest(
454
455
456
# TODO : deprecate _llama_gemma_update_causal_mask_legacy when transformers>=4.41.0
457
-if is_transformers_version(">=", "4.41.0"):
+if is_transformers_version(">", "4.40.0"):
458
_llama_gemma_update_causal_mask = _llama_gemma_update_causal_mask_latest
459
else:
460
_llama_gemma_update_causal_mask = _llama_gemma_update_causal_mask_legacy
0 commit comments