-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
31 lines (27 loc) · 670 Bytes
/
pyproject.toml
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
[tool.poetry]
name = "yi-models"
version = "0.1.1-dev"
description = ""
authors = ["Yi Team <[email protected]>"]
license = "Apache 2.0"
readme = "README.md"
[tool.poetry.dependencies]
python = "~3.10"
pytorch = "2.0.1"
pytorch-cuda = "11.8"
deepspeed = "0.12.2"
transformers = "4.36.0"
sentencepiece = "0.1.99"
accelerate = "0.24.1"
datasets = "2.14.5"
optimum = "1.13.2"
einops = "0.7.0"
flash-attn = {version = "2.3.6", 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"