Skip to content

Add pytree registration for static cache#41754

Draft
angelayi wants to merge 1 commit intohuggingface:mainfrom
angelayi:angelayi/pytree_static_Cache
Draft

Add pytree registration for static cache#41754
angelayi wants to merge 1 commit intohuggingface:mainfrom
angelayi:angelayi/pytree_static_Cache

Conversation

@angelayi
Copy link
Copy Markdown
Contributor

@angelayi angelayi commented Oct 21, 2025

  • Refactored the existing pytree registration for DynamicCache to be a more generalized form
  • Also fixed the existing pytree registration which triggers lazy_initialization. This shouldn't be needed because we already have all the contents of the dynamic cache created?
  • Added pytree support for StaticCache

cc @tugsbayasgalan

Copy link
Copy Markdown
Collaborator

@ArthurZucker ArthurZucker left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Happy to merge at your convenience! Sorry missed the notif

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants