fix i

Author: xadupre

onnx_diagnostic/torch_export_patches/onnx_export_errors.py

@@ -68,7 +68,6 @@ def unpatch_module(mod, info: Dict[type, Dict[type, Callable]], verbose: int = 0
 def _register_cache_serialization(verbose: int = 0) -> Dict[str, bool]:
     # Cache serialization: to be moved into appropriate packages
     import torch
-    import transformers
     import packaging.version as pv
 
     try:
@@ -111,7 +110,7 @@ def _register_cache_serialization(verbose: int = 0) -> Dict[str, bool]:
     #           DynamicCache, _flatten_dynamic_cache_for_fx)
     # so we remove it anyway
     if DynamicCache in torch.fx._pytree.SUPPORTED_NODES and pv.Version(
-        transformers.__version__
+        torch.__version__
     ) >= pv.Version("2.7"):
         if verbose:
             print("[_register_cache_serialization] DynamicCache is unregistered first.")