extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-24 17:24:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Bring back the ability to run on CPU on PCs with CUDA-compatible GPUs

Browse Source
This commit is contained in:
cmdr2 2023-03-10 10:19:55 +05:30
parent 8907dabd4c
commit 0fd706f392
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ui/easydiffusion/device_manager.py
View File

@ -65,10 +65,12 @@ def get_device_delta(render_devices, active_devices):
def is_mps_available():
return platform.system() == "Darwin" and \
hasattr(torch.backends, 'mps') and \
torch.backends.mps.is_available() and \
torch.backends.mps.is_built()
return (
platform.system() == "Darwin"
and hasattr(torch.backends, "mps")
and torch.backends.mps.is_available()
and torch.backends.mps.is_built()
)
def is_cuda_available():
@ -213,7 +215,7 @@ def is_device_compatible(device):
log.error(str(e))
return False
if not is_cuda_available():
if device in ("cpu", "mps"):
return True
# Memory check
try:

2
ui/easydiffusion/task_manager.py
View File

@ -385,7 +385,7 @@ def get_devices():
}
def get_device_info(device):
if not device_manager.is_cuda_available():
if device in ("cpu", "mps"):
return {"name": device_manager.get_processor_name()}
mem_free, mem_total = torch.cuda.mem_get_info(device)