aboutsummaryrefslogtreecommitdiff
path: root/modules/mac_specific.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-09 14:40:06 +0300
committerGitHub <noreply@github.com>2023-08-09 14:40:06 +0300
commitc8c48640e661a0275c9ffa04b148e20213f3e902 (patch)
tree601a7db6f6811ce6c0a1cb02e7d9c30b705595bb /modules/mac_specific.py
parent2617598b7a014676611134566fafd6604e6c4486 (diff)
parent386245a26427a64f364f66f6fecd03b3bccfd7f3 (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.py4
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