Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 8a580dd

Browse files
authored
Add MPS Fallback to support Torchtune Multimodal MPS (#1241)
1 parent 8c7e688 commit 8a580dd

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

torchchat.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,11 @@
99
import subprocess
1010
import sys
1111

12+
# MPS ops missing with Multimodal torchtune
13+
# https://github.com/pytorch/torchtune/issues/1723
14+
import os
15+
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
16+
1217
from torchchat.cli.cli import (
1318
add_arguments_for_verb,
1419
arg_init,

0 commit comments

Comments
 (0)