diff options
author | Aarni Koskela <akx@iki.fi> | 2023-07-12 15:17:13 +0300 |
---|---|---|
committer | Aarni Koskela <akx@iki.fi> | 2023-07-12 15:17:30 +0300 |
commit | 3d524fd3f1bdb17946bf6fa8a3cdf7b10859c495 (patch) | |
tree | 8f7b7906d1257ddfe71aa56427d1103d7ba10c26 /modules/mac_specific.py | |
parent | 8f6b24ce5922174d96eb9776126488cb28694ff8 (diff) |
Don't do MPS GC when there's a latent that could still be sampled
Diffstat (limited to 'modules/mac_specific.py')
-rw-r--r-- | modules/mac_specific.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/mac_specific.py b/modules/mac_specific.py index 328b5973..9ceb43ba 100644 --- a/modules/mac_specific.py +++ b/modules/mac_specific.py @@ -30,6 +30,10 @@ has_mps = check_for_mps() def torch_mps_gc() -> None: try: + from modules.shared import state + if state.current_latent is not None: + log.debug("`current_latent` is set, skipping MPS garbage collection") + return from torch.mps import empty_cache empty_cache() except Exception: |