diffusers
c1dc2ae6 - Fix multi-gpu case for `train_cm_ct_unconditional.py` (#8653)

Commit
1 year ago
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()`
Author
Parents
Loading