fix mps miss import

3e2309f1 · KarryCharon · 2b2a1474 · 3e2309f1
Commit 3e2309f1 authored Jul 12, 2023 by KarryCharon
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

comfy/model_management.py comfy/model_management.py +1 -0

No files found.
--- 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