File size: 766 Bytes
3959ccd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
[tool.poetry]
name = "yi-models"
version = "0.1.0-dev"
description = ""
authors = ["Yi Team <[email protected]>"]
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"
|