Skip to content

Commit 6821e42

Browse files
committed
fix
1 parent c9d1e52 commit 6821e42

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/cache_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -585,7 +585,7 @@ def _flatten_dynamic_cache_for_fx(cache, spec):
585585
return torch.utils._pytree.tree_flatten(dictionary)[0]
586586

587587

588-
if is_torch_greater_or_equal("2.2"):
588+
if is_torch_greater_or_equal("2.3"):
589589
torch.utils._pytree.register_pytree_node(
590590
DynamicCache,
591591
_flatten_dynamic_cache,

0 commit comments

Comments
 (0)