Relax triton requirements for compatibility with pytorch 2.1 and newer (

openai#1802)

setup.py

@@ -14,7 +14,6 @@ def read_version(fname="whisper/version.py"):
 requirements = []
 whisper_rocm = os.getenv('WHISPER_ROCM',default='0')
 if sys.platform.startswith("linux") and platform.machine() == "x86_64":
-<<<<<<< HEAD
  from check_rocm_platform import is_command, check_amd_gpu_rocminfo, check_amd_gpu_lspci, check_rocm_packages
  ROCM_PLATFORM = False
  if is_command("rocminfo"):
@@ -30,14 +29,8 @@ def read_version(fname="whisper/version.py"):
  print("rocm")
  requirements.append("pytorch-triton-rocm>=2.0.1")
  else :
- requirements.append("triton==2.0.0")
-=======
- if whisper_rocm == "1" :
- requirements.append("pytorch_triton_rocm==2.0.2")
- else :
- requirements.append("triton==2.0.0")
+ requirements.append("triton>=2.0.0,<3")
 
->>>>>>> c41de6ac767b30524779e260969ed4a4d6a43523
 setup(
  name="openai-whisper",
  py_modules=["whisper"],