Add MPS Fallback to support Torchtune Multimodal MPS (#1241)

pytorch · Oct 1, 2024 · 8a580dd · 8a580dd
1 parent 8c7e688
commit 8a580dd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/torchchat.py b/torchchat.py
@@ -9,6 +9,11 @@
 import subprocess
 import sys
 
+# MPS ops missing with Multimodal torchtune
+# https://github.com/pytorch/torchtune/issues/1723
+import os
+os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
+
 from torchchat.cli.cli import (
     add_arguments_for_verb,
     arg_init,