Loading modules/mac_specific.py +5 −1 Original line number Diff line number Diff line Loading @@ -5,7 +5,7 @@ import platform from modules.sd_hijack_utils import CondFunc from packaging import version log = logging.getLogger() log = logging.getLogger(__name__) # before torch version 1.13, has_mps is only available in nightly pytorch and macOS 12.3+, Loading @@ -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: Loading Loading
modules/mac_specific.py +5 −1 Original line number Diff line number Diff line Loading @@ -5,7 +5,7 @@ import platform from modules.sd_hijack_utils import CondFunc from packaging import version log = logging.getLogger() log = logging.getLogger(__name__) # before torch version 1.13, has_mps is only available in nightly pytorch and macOS 12.3+, Loading @@ -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: Loading
