Fix multi-gpu case for `train_cm_ct_unconditional.py` (#8653)
* Fix multi-gpu case
* Prefer previously created `unwrap_model()` function
For `torch.compile()` generalizability
* `chore: update unwrap_model() function to use accelerator.unwrap_model()`