diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-09 14:40:06 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-09 14:40:06 +0300 |
commit | c8c48640e661a0275c9ffa04b148e20213f3e902 (patch) | |
tree | 601a7db6f6811ce6c0a1cb02e7d9c30b705595bb /modules/mac_specific.py | |
parent | 2617598b7a014676611134566fafd6604e6c4486 (diff) | |
parent | 386245a26427a64f364f66f6fecd03b3bccfd7f3 (diff) |
Merge pull request #12426 from AUTOMATIC1111/split_shared
Split shared.py into multiple files
Diffstat (limited to 'modules/mac_specific.py')
-rw-r--r-- | modules/mac_specific.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/mac_specific.py b/modules/mac_specific.py index 9ceb43ba..bce527cc 100644 --- a/modules/mac_specific.py +++ b/modules/mac_specific.py @@ -4,6 +4,7 @@ import torch import platform from modules.sd_hijack_utils import CondFunc from packaging import version +from modules import shared log = logging.getLogger(__name__) @@ -30,8 +31,7 @@ has_mps = check_for_mps() def torch_mps_gc() -> None: try: - from modules.shared import state - if state.current_latent is not None: + if shared.state.current_latent is not None: log.debug("`current_latent` is set, skipping MPS garbage collection") return from torch.mps import empty_cache |