Skipping pytree registration in case fsdp is enabled#40075

Cyrilvallez merged 19 commits intohuggingface:mainhuggingface/transformers:mainfrom

romitjain:bugfix/fsdp-pytree-memory-regressionromitjain/transformers:bugfix/fsdp-pytree-memory-regressionCopy head branch name to clipboard