File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
onnx_diagnostic/torch_export_patches Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -492,7 +492,11 @@ def _patch_transformers(
492492 except ImportError :
493493 modeling_rope_utils = None
494494
495- if patch_details and modeling_rope_utils :
495+ if (
496+ patch_details
497+ and modeling_rope_utils
498+ and hasattr (modeling_rope_utils , "dynamic_rope_update" )
499+ ):
496500 patch_details .append (
497501 "patch_transformers" ,
498502 modeling_rope_utils .dynamic_rope_update ,
Original file line number Diff line number Diff line change @@ -66,7 +66,7 @@ def _is_torchdynamo_exporting() -> bool:
6666 return False
6767
6868
69- patch_sdpa_is_causal = _has_transformers ("4.55 " )
69+ patch_sdpa_is_causal = _has_transformers ("5.0 " )
7070patch_is_initialized = _has_transformers ("4.56.99" )
7171
7272
You can’t perform that action at this time.
0 commit comments