accelerate

02a8a9a3 - Fix test_script.py on TPU v2/v3 (#2542)

Fix test_script.py on TPU v2/v3 (#2542) * fix replication * Set generator on each thread. The test passed. * remove comments * fix up * fix format * fix comment * not setting the dataloader.batch_sampler