diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-10 17:04:38 +0300 |
---|---|---|
committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-10 17:04:38 +0300 |
commit | 70a01cd4440d708bf25cc50393c0430935a8ebc2 (patch) | |
tree | daad07800a3dadfd3caeac1383c1c65ecfcb6284 /modules/mac_specific.py | |
parent | 1aefb5025929818b2a96cbb6148fcc2db7b947ec (diff) | |
parent | 070b034cd5b49eb5056a18b43f88aa223fec9e0b (diff) |
Merge branch 'dev' into refiner
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 |