Merge pull request #448 from peanutcocktail/main

Set PYTORCH_ENABLE_MPS_FALLBACK=1 for MPS by default
2026-07-28 13:26:59 -07:00 · 2024-11-11 07:11:19 +08:00
parent 3fcdbc70b4 5635eac12d
commit e6e7bd01d3
1 changed files with 2 additions and 0 deletions
@@ -32,6 +32,8 @@ from f5_tts.model.utils import (
 _ref_audio_cache = {}

 device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
+if device == "mps":
+    os.environ["PYTOCH_ENABLE_MPS_FALLBACK"] = "1"

 # -----------------------------------------