From d89746f56ad06c15e69149b4d1dbb5a3bdab72ba Mon Sep 17 00:00:00 2001 From: roy Date: Sun, 14 Apr 2024 12:19:03 +0800 Subject: [PATCH 1/2] upgrade --- requirements-cpu.txt | 2 +- requirements-cuda.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements-cpu.txt b/requirements-cpu.txt index 5779b38b24e69..e911ad03295f0 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -3,4 +3,4 @@ # Dependencies for x86_64 CPUs torch == 2.2.1+cpu -triton >= 2.1.0 # FIXME(woosuk): This is a hack to avoid import error. \ No newline at end of file +triton >= 2.2.0 # FIXME(woosuk): This is a hack to avoid import error. \ No newline at end of file diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 6ee75e8139c04..d28b5eea29756 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -7,4 +7,4 @@ pynvml == 11.5.0 vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library torch == 2.2.1 xformers == 0.0.25 # Requires PyTorch 2.2.1 -triton >= 2.1.0 +triton >= 2.2.0 From 297f902b638508fd6ba56c7650e3e1314df393bc Mon Sep 17 00:00:00 2001 From: roy Date: Sun, 14 Apr 2024 12:31:23 +0800 Subject: [PATCH 2/2] remove triton --- requirements-cuda.txt | 1 - 1 file changed, 1 deletion(-) diff --git a/requirements-cuda.txt b/requirements-cuda.txt index d28b5eea29756..c6d2cd46aee54 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -7,4 +7,3 @@ pynvml == 11.5.0 vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library torch == 2.2.1 xformers == 0.0.25 # Requires PyTorch 2.2.1 -triton >= 2.2.0