fix mps miss import

2 years ago · 3e2309f149
parent 2b2a1474f7
commit 3e2309f149
1 changed files with 1 additions and 0 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -49,6 +49,7 @@ except:
 try:
    if torch.backends.mps.is_available():
        cpu_state = CPUState.MPS
+        import torch.mps
 except:
    pass