diff options
| author | Matthew McGoogan <mlmcgoogan@gmail.com> | 2022-11-26 23:25:16 +0000 |
|---|---|---|
| committer | Matthew McGoogan <mlmcgoogan@gmail.com> | 2022-11-26 23:25:16 +0000 |
| commit | c67c40f983997594f76b2312f92c3761e8d83715 (patch) | |
| tree | 8f040a774bbde068b374d71a6fc1467be0195d14 /modules/sd_models.py | |
| parent | b5050ad2071644f7b4c99660dc66a8a95136102f (diff) | |
torch.cuda.empty_cache() defaults to cuda:0 device unless explicitly set otherwise first. Updating torch_gc() to use the device set by --device-id if specified to avoid OOM edge cases on multi-GPU systems.
Diffstat (limited to 'modules/sd_models.py')
0 files changed, 0 insertions, 0 deletions