diodiogod-TTS-Audio-Suite/utils/device/__init__.py at codex/audio-cpp-integration - diodiogod-TTS-Audio-Suite - Gitea: Git with a cup of tea

nodes/diodiogod-TTS-Audio-Suite

Files

T

diodiogod 573c2497d3 Add MPS (Apple Silicon) support for all TTS engines

- Created utils/device/torch_device_resolver.py with intelligent device detection
  - Supports MPS (Metal Performance Shaders) for Apple Silicon
  - Fallback chain: MPS > CUDA > CPU for auto-detection
  - Safe detection with torch.backends.mps.is_available() checks

- Updated all 7 engine nodes to expose 'mps' device option:
  - ChatterBox, F5-TTS, Higgs Audio, VibeVoice
  - ChatterBox Official 23-Lang, Index TTS, RVC

- Integrated device resolver across all engine implementations:
  - unified_model_interface.py (central point)
  - comfyui_model_wrapper/base_wrapper.py (ComfyUI integration)
  - All direct engine implementations (VibeVoice, Higgs Audio, F5-TTS edit, etc.)

- Maintains backward compatibility:
  - Existing workflows unchanged
  - MPS added as final option in device dropdowns
  - No behavior changes for Linux/Windows users

Resolves #160 - Users on Apple Silicon can now select MPS for GPU acceleration

2025-11-04 19:13:57 -03:00

6 lines

203 B

Python

Raw Permalink Blame History

 """Device utilities for TTS Audio Suite"""
 from utils.device.torch_device_resolver import resolve_torch_device, validate_device_string
 __all__ = ["resolve_torch_device", "validate_device_string"]