fix mps miss import

2023-07-12 10:06:34 +08:00 · 2023-07-12 10:06:34 +08:00 · 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