init

2026-01-09 13:34:11 +08:00
parent dfa6476b58
commit b2ef04d792
538 changed files with 105693 additions and 2 deletions
--- a/requirements-cuda.txt
+++ b/requirements-cuda.txt
@@ -0,0 +1,9 @@
+# Common dependencies
+-r requirements-common.txt
+
+# Dependencies for NVIDIA GPUs
+ray >= 2.9
+nvidia-ml-py # for pynvml package
+vllm-nccl-cu12>=2.18,<2.19  # for downloading nccl library
+torch == 2.3.0
+xformers == 0.0.26.post1  # Requires PyTorch 2.3.0