Skip to content

Commit 6a34e81

Browse files
committed
fix ache
1 parent de5958e commit 6a34e81

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,13 @@
4343

4444
KWARGS_LAYER_NAMES = {
4545
"DynamicLayer": lambda layer: "",
46-
"DynamicSlidingWindowLayer": lambda layer: str(layer.sliding_window),
46+
"DynamicSlidingWindowLayer": lambda layer: str(
47+
getattr(layer, "sliding_window", getattr(layer, "max_cache_len", 0))
48+
),
4749
"StaticLayer": lambda layer: "",
48-
"StaticSlidingWindowLayer": lambda layer: str(layer.sliding_window),
50+
"StaticSlidingWindowLayer": lambda layer: str(
51+
getattr(layer, "sliding_window", getattr(layer, "max_cache_len", 0))
52+
),
4953
}
5054

5155
PARSE_LAYER_NAMES = {

0 commit comments

Comments
 (0)