accelerate

c2a16293 - Fix nested context manager for main_process_first() (#1304)

Fix nested context manager for main_process_first() (#1304) * Fix nested context manager for main_process_first() * Fix test for main_process_first() * Improve test for main_process_first() * Fix formatting * Fix test with single process