Skipping pytree registration in case fsdp is enabled #40075
Cyrilvallez
merged
19 commits into
huggingface:main
from
romitjain:bugfix/fsdp-pytree-memory-regression
Skipping pytree registration in case fsdp is enabled
4d36ce3c
Beauty changes
41363a84
Beauty changes
16e9c5c4
Merge branch 'main' into bugfix/fsdp-pytree-memory-regression
64b8242c
gante
approved these changes
on 2025-08-12
Moved the is_fsdp_available function to import utils
763fd3f4
Moved is_fsdp_available to integrations.fsdp
5c723f30
Skipping pytree registration in case fsdp is enabled
0a30018e
Beauty changes
66bb47c0
Beauty changes
83c37ff8
Moved the is_fsdp_available function to import utils
5b7c3848
Moved is_fsdp_available to integrations.fsdp
5c3b0182
Added pytree registration inside dynamic cache class
f4213d4e
Making ci/cd lords happy
8edb59dc
Adding a check if DynamicCache is already a leaf
89cd656d
Adding try/catch for multiple initializations of DynamicCache in test…
6c861a1a
Merge branch 'bugfix/fsdp-pytree-memory-regression' of github.com:rom…
0e6907c3
Merge branch 'bugfix/fsdp-pytree-memory-regression' of github.com:rom…
fd302075
Moving dynamic cache pytree registration to executorch
3b558fcc
Adding try catch back
93c9cf25
gante
approved these changes
on 2025-08-19
Cyrilvallez
approved these changes
on 2025-08-19
SunMarc
commented
on 2025-08-20
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub