From 3ee78c064bada09f2d0a36c91a8df0d2b6e6d988 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 e148408b8..fc2c572e2 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