Disable triton via env when platform is Windows

main v1.17.0
Uminosachi 2024-06-22 16:26:51 +09:00
parent ac2da7e9bf
commit 03a1fabb16
1 changed files with 3 additions and 0 deletions

View File

@ -6,6 +6,9 @@ import platform
if platform.system() == "Darwin":
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
if platform.system() == "Windows":
os.environ["XFORMERS_FORCE_DISABLE_TRITON"] = "1"
import random
import re
import traceback