01-ai-bot
commited on
Commit
•
3959ccd
1
Parent(s):
3ec17ea
Sync files
Browse files- pyproject.toml +33 -0
pyproject.toml
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[tool.poetry]
|
| 2 |
+
name = "yi-models"
|
| 3 |
+
version = "0.1.0-dev"
|
| 4 |
+
description = ""
|
| 5 |
+
authors = ["Yi Team <yi@01.ai>"]
|
| 6 |
+
license = "Apache 2.0"
|
| 7 |
+
readme = "README.md"
|
| 8 |
+
|
| 9 |
+
[tool.poetry.dependencies]
|
| 10 |
+
python = "^3.10"
|
| 11 |
+
pytorch = "2.1.0"
|
| 12 |
+
pytorch-cuda = "11.8"
|
| 13 |
+
deepspeed = "0.12.2"
|
| 14 |
+
transformers = "4.34.0"
|
| 15 |
+
sentencepiece = "0.1.99"
|
| 16 |
+
accelerate = "0.24.1"
|
| 17 |
+
datasets = "2.14.5"
|
| 18 |
+
optimum = "1.13.2"
|
| 19 |
+
einops = "0.7.0"
|
| 20 |
+
exllama = {version = "0.1.0", source = "pypi"}
|
| 21 |
+
auto-gptq = {version = "0.4.2", source = "pypi"}
|
| 22 |
+
flash-attn = {version = "2.3.3", source = "pypi"}
|
| 23 |
+
|
| 24 |
+
[tool.conda-lock]
|
| 25 |
+
channels = ["conda-forge", "pytorch", "nvidia", "nodefaults"]
|
| 26 |
+
platforms = ["linux-64"]
|
| 27 |
+
|
| 28 |
+
[tool.conda-lock.dependencies]
|
| 29 |
+
pip = ">=23.3"
|
| 30 |
+
|
| 31 |
+
[build-system]
|
| 32 |
+
requires = ["poetry-core"]
|
| 33 |
+
build-backend = "poetry.core.masonry.api"
|