Try installing triton only if linux & x86_64 (#1051)
This commit is contained in:
5
setup.py
5
setup.py
@@ -1,4 +1,5 @@
|
|||||||
import os
|
import os
|
||||||
|
import platform
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
import pkg_resources
|
import pkg_resources
|
||||||
@@ -11,8 +12,8 @@ def read_version(fname="whisper/version.py"):
|
|||||||
|
|
||||||
|
|
||||||
requirements = []
|
requirements = []
|
||||||
if sys.platform.startswith("linux"):
|
if sys.platform.startswith("linux") and platform.machine() == "x86_64":
|
||||||
triton_requirement = "triton>=2.0.0.dev20221202"
|
triton_requirement = "triton==2.0.0"
|
||||||
try:
|
try:
|
||||||
import re
|
import re
|
||||||
import subprocess
|
import subprocess
|
||||||
|
|||||||
Reference in New Issue
Block a user