Skip to content

Commit efd7b5e

Browse files
committed
fix issues
1 parent a4d323c commit efd7b5e

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -846,7 +846,7 @@ class patched_GemmaRotaryEmbedding(common_RotaryEmbedding):
846846
_PATCHED_CLASS_ = transformers.models.gemma.modeling_gemma.GemmaRotaryEmbedding
847847

848848

849-
if pv.Version(transformers.__version__) >= pv.Version("4.53"):
849+
if pv.Version(transformers.__version__) >= pv.Version("4.52"):
850850

851851
class patched_Gemma2RotaryEmbedding(common_RotaryEmbedding):
852852
_PATCHES_ = ["forward"]
@@ -877,19 +877,17 @@ class patched_PhiRotaryEmbedding(common_RotaryEmbedding):
877877
_PATCHED_CLASS_ = transformers.models.phi.modeling_phi.PhiRotaryEmbedding
878878

879879

880-
class patched_Phi3RotaryEmbedding(common_RotaryEmbedding):
881-
_PATCHES_ = ["forward"]
882-
_PATCHED_CLASS_ = transformers.models.phi3.modeling_phi3.Phi3RotaryEmbedding
883-
884-
885-
class patched_Phi4MultimodalRotaryEmbedding(common_RotaryEmbedding):
886-
_PATCHES_ = ["forward"]
887-
_PATCHED_CLASS_ = (
888-
transformers.models.phi4_multimodal.modeling_phi4_multimodal.Phi4MultimodalRotaryEmbedding
889-
)
880+
if pv.Version(transformers.__version__) >= pv.Version("4.52"):
890881

882+
class patched_Phi3RotaryEmbedding(common_RotaryEmbedding):
883+
_PATCHES_ = ["forward"]
884+
_PATCHED_CLASS_ = transformers.models.phi3.modeling_phi3.Phi3RotaryEmbedding
891885

892-
if pv.Version(transformers.__version__) >= pv.Version("4.53"):
886+
class patched_Phi4MultimodalRotaryEmbedding(common_RotaryEmbedding):
887+
_PATCHES_ = ["forward"]
888+
_PATCHED_CLASS_ = (
889+
transformers.models.phi4_multimodal.modeling_phi4_multimodal.Phi4MultimodalRotaryEmbedding
890+
)
893891

894892
class patched_SmolLM3RotaryEmbedding(common_RotaryEmbedding):
895893
_PATCHES_ = ["forward"]

0 commit comments

Comments
 (0)