Fix default world_size when running on 1 or 0 GPU (#119372)
the mentioned distributed tests would fail if the number of GPUs available isn't sufficient. need to correct the default world size.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/119372
Approved by: https://github.com/eqy, https://github.com/fegin