Skip to content

Commit b3c411f

Browse files
committed
fix
1 parent 6d6dedb commit b3c411f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

onnx_diagnostic/helpers/torch_helper.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -699,6 +699,7 @@ def to_any(value: Any, to_value: Union[torch.dtype, torch.device, str]) -> Any:
699699
)
700700
and hasattr(value, "dtype")
701701
and value.dtype in {torch.int32, torch.int64, torch.int8, torch.int16}
702+
and value.__class__.__name__ not in {"DynamicCache", "EncoderDecoderCache"}
702703
):
703704
# int vector should not be changed.
704705
return value

0 commit comments

Comments
 (0)