Merge pull request #136 from qidanrui/codellama-7b-result

update torch version to ,add dependency
eosphoros-ai · Nov 14, 2023 · d8d8cae · d8d8cae
2 parents 5b7e567 + fc68110
commit d8d8cae
Show file tree

Hide file tree

Showing 2 changed files with 182 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,6 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
 transformers = "4.33.2"
-torch = ">=1.13.1"
 peft = "0.4.0"
 accelerate = "0.21.0"
 bitsandbytes = "0.39.0"
@@ -43,6 +42,21 @@ pyyaml = "6.0.1"
 black = "^23.11.0"
 pyright = "^1.1.335"
 pylint = "^3.0.2"
+torch = "^2.0.0"
+markupsafe = "2.1.3"
+nvidia-cuda-nvrtc-cu12 = "12.1.105"
+nvidia-cuda-runtime-cu12 = "12.1.105"
+nvidia-cuda-cupti-cu12 = "12.1.105"
+nvidia-cudnn-cu12 = "8.9.2.26"
+nvidia-cublas-cu12 = "12.1.3.1"
+nvidia-cufft-cu12 = "11.0.2.54"
+nvidia-curand-cu12 = "10.3.2.106"
+nvidia-cusolver-cu12 = "11.4.5.107"
+nvidia-cusparse-cu12 = "12.1.0.106"
+nvidia-nccl-cu12 = "2.18.1"
+nvidia-nvtx-cu12 = "12.1.105"
+triton = "2.1.0"
+nvidia-nvjitlink-cu12 = "^12.3.52"
 
 [build-system]
 requires = ["poetry-core"]