Update MPS high watermark ratio to 1.9 on macOS

zackees · Jun 22, 2024 · 90fd75b · 90fd75b
1 parent 1225d7c
commit 90fd75b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transcribe_anything/insanely_fast_whisper.py b/transcribe_anything/insanely_fast_whisper.py
@@ -202,7 +202,7 @@ def run_insanely_fast_whisper(
     if sys.platform == "darwin":
         # Attempts fixed recommended for the mps machines. This seems
         # to be necessary since a recent update.
-        env["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
+        env["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "1.9"
         env["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
     device_id = get_device_id()
     cmd_list = []