Merge pull request #11749 from akx/mps-gc-fix-2

Don't do MPS GC when there's a latent

modules/mac_specific.py

@@ -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+,
@@ -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: