accelerate

2b1e7bd4 - Fixup `free_memory` to deal with garbage collection (#2716)

Fixup `free_memory` to deal with garbage collection (#2716) * Fixup cleanup * Return * Fixup test * Fix test * DeepSpeed * More careful guard * bring back as none * passing * bring forward