Repository navigation

Skipping pytree registration in case fsdp is enabled - #40075

Merged

Cyrilvallez merged 19 commits into

huggingface:mainfrom

romitjain:bugfix/fsdp-pytree-memory-regression

Aug 19, 2025

Merged

Skipping pytree registration in case fsdp is enabled#40075
Cyrilvallez merged 19 commits into
huggingface:mainfrom
romitjain:bugfix/fsdp-pytree-memory-regression

Commits on Aug 11, 2025
- Skipping pytree registration in case fsdp is enabled
  romitjain
  committed
  4d36ce3
- Beauty changes
  romitjain
  committed
  41363a8
- Beauty changes
  romitjain
  committed
  16e9c5c
Commits on Aug 12, 2025
- Merge branch 'main' into bugfix/fsdp-pytree-memory-regression
  romitjain
  authored
  64b8242
Commits on Aug 13, 2025
- Moved the is_fsdp_available function to import utils
  romitjain
  committed
  763fd3f
- Moved is_fsdp_available to integrations.fsdp
  romitjain
  committed
  5c723f3
Commits on Aug 14, 2025
- Skipping pytree registration in case fsdp is enabled
  romitjain
  committed
  0a30018
- Beauty changes
  romitjain
  committed
  66bb47c
- Beauty changes
  romitjain
  committed
  83c37ff
- Moved the is_fsdp_available function to import utils
  romitjain
  committed
  5b7c384
- Moved is_fsdp_available to integrations.fsdp
  romitjain
  committed
  5c3b018
- Added pytree registration inside dynamic cache class
  romitjain
  committed
  f4213d4
- Making ci/cd lords happy
  romitjain
  committed
  8edb59d
- Adding a check if DynamicCache is already a leaf
  romitjain
  committed
  89cd656
- Adding try/catch for multiple initializations of DynamicCache in test suites
  romitjain
  committed
  6c861a1
Commits on Aug 19, 2025