Skip to content

Commit 4092066

Browse files
committed
fix a few things
1 parent 7a098b9 commit 4092066

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

onnx_diagnostic/torch_export_patches/onnx_export_serialization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ def serialization_functions(
231231
)
232232

233233
SlidingWindowCache = get_sliding_window_cache_cls()
234-
if SlidingWindowCache:
234+
if SlidingWindowCache and SlidingWindowCache.__name__ == "SlidingWindowCache":
235235
from .serialization.transformers_impl import (
236236
flatten_sliding_window_cache,
237237
unflatten_sliding_window_cache,

onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ def unflatten_static_cache(
220220
####################
221221

222222

223-
if SlidingWindowCache:
223+
if SlidingWindowCache and SlidingWindowCache.__name__ == "SlidingWindowCache":
224224

225225
def flatten_sliding_window_cache(
226226
cache: SlidingWindowCache,

0 commit comments

Comments
 (0)