Update PYTORCH_MPS_HIGH_WATERMARK_RATIO to 1.9

zackees · Jun 22, 2024 · cc4c37a · cc4c37a
1 parent e245d0f
commit cc4c37a
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"
+        env["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "1.9"
     device_id = get_device_id()
     cmd_list = []
     output_dir.mkdir(parents=True, exist_ok=True)