Dymas/abogen
1
0
Fork 0
mirror of https://github.com/denizsafak/abogen.git synced 2026-07-18 13:40:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Test MPS GPU acceleratin on Silicon Mac

Browse Source
This commit is contained in:
Deniz Şafak
2025-08-07 05:22:43 +03:00
parent 395cd75038
commit e48df7c106
2 changed files with 35 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
abogen/utils.py
+15 -5
View File
@@ -239,13 +239,23 @@ def calculate_text_length(text):
def get_gpu_acceleration(enabled):
try:
import torch
from torch.cuda import is_available
from torch.cuda import is_available as cuda_available
if not enabled:
return "CUDA GPU available but using CPU.", False
if is_available():
return "GPU available but using CPU.", False
# Check for Apple Silicon MPS
if platform.system() == "Darwin" and platform.processor() == "arm":
if torch.backends.mps.is_available():
return "MPS GPU available and enabled.", True
else:
return "MPS GPU not available on Apple Silicon. Using CPU.", False
# Check for CUDA
if cuda_available():
return "CUDA GPU available and enabled.", True
# Gather more diagnostic info if CUDA is not available
# Gather CUDA diagnostic info if not available
try:
cuda_devices = torch.cuda.device_count()
cuda_error = (
@@ -257,7 +267,7 @@ def get_gpu_acceleration(enabled):
cuda_error = str(e)
return f"CUDA GPU is not available. Using CPU. ({cuda_error})", False
except Exception as e:
return f"Error checking CUDA GPU: {e}", False
return f"Error checking GPU: {e}", False
def prevent_sleep_start():