| [tool.poetry] | |
| name = "yi-models" | |
| version = "0.1.0-dev" | |
| description = "" | |
| authors = ["Yi Team <yi@01.ai>"] | |
| license = "Apache 2.0" | |
| readme = "README.md" | |
| [tool.poetry.dependencies] | |
| python = "^3.10" | |
| pytorch = "2.1.0" | |
| pytorch-cuda = "11.8" | |
| deepspeed = "0.12.2" | |
| transformers = "4.34.0" | |
| sentencepiece = "0.1.99" | |
| accelerate = "0.24.1" | |
| datasets = "2.14.5" | |
| optimum = "1.13.2" | |
| einops = "0.7.0" | |
| exllama = {version = "0.1.0", source = "pypi"} | |
| auto-gptq = {version = "0.4.2", source = "pypi"} | |
| flash-attn = {version = "2.3.3", source = "pypi"} | |
| [tool.conda-lock] | |
| channels = ["conda-forge", "pytorch", "nvidia", "nodefaults"] | |
| platforms = ["linux-64"] | |
| [tool.conda-lock.dependencies] | |
| pip = ">=23.3" | |
| [build-system] | |
| requires = ["poetry-core"] | |
| build-backend = "poetry.core.masonry.api" | |