File tree Expand file tree Collapse file tree 1 file changed +1
-15
lines changed
Expand file tree Collapse file tree 1 file changed +1
-15
lines changed Original file line number Diff line number Diff line change @@ -13,21 +13,7 @@ def read_version(fname="whisper/version.py"):
1313
1414requirements = []
1515if sys .platform .startswith ("linux" ) and platform .machine () == "x86_64" :
16- triton_requirement = "triton==2.0.0"
17- try :
18- import re
19- import subprocess
20-
21- version_line = (
22- subprocess .check_output (["nvcc" , "--version" ]).strip ().split (b"\n " )[- 1 ]
23- )
24- major , minor = re .findall (rb"([\d]+)\.([\d]+)" , version_line )[0 ]
25- if (int (major ), int (minor )) < (11 , 4 ):
26- # the last version supporting CUDA < 11.4
27- triton_requirement = "triton==2.0.0.dev20221011"
28- except (IndexError , OSError , subprocess .SubprocessError ):
29- pass
30- requirements .append (triton_requirement )
16+ requirements .append ("triton==2.0.0" )
3117
3218setup (
3319 name = "openai-whisper" ,
You can’t perform that action at this time.
0 commit comments