Skip to content

GitLab

Commit 3e2309f1 authored by KarryCharon's avatar KarryCharon
Browse files

fix mps miss import

parent 2b2a1474
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy/model_management.py
View file @ 3e2309f1
...@@ -49,6 +49,7 @@ except: ...@@ -49,6 +49,7 @@ except:
try: try:
if torch.backends.mps.is_available(): if torch.backends.mps.is_available():
cpu_state = CPUState.MPS cpu_state = CPUState.MPS
import torch.mps
except: except:
pass pass
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment