From 3e2309f149c4f6d3ef8722720b7fb67109349fed Mon Sep 17 00:00:00 2001
From: KarryCharon <karrycharon@qq.com>
Date: Wed, 12 Jul 2023 10:06:34 +0800
Subject: [PATCH] fix mps miss import

---
 comfy/model_management.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index e148408b..fc2c572e 100644
--- 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