Merge branch 'fix-AttributeError-module-'torch'-has-no-attribute-'mps'' of https://github.com/KarryCharon/ComfyUI

2023-07-20 00:34:54 -04:00 · 2023-07-20 00:34:54 -04:00 · 4760c29380
parent ccb6b70de1 3e2309f149
commit 4760c29380
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