forked from CambioML/pykoi-rlhf-finetuned-transformers
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpyproject.toml
47 lines (43 loc) · 918 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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
[tool.poetry]
name = "pykoi"
version = "0.0.5"
description = ""
authors = ["CambioML <[email protected]>"]
license = "Apache"
readme = "README.md"
[tool.poetry.dependencies]
python = ">=3.9,<3.13"
fastapi = "0.100.0"
pydantic = "1.10.11"
starlette = "0.27.0"
uvicorn = "0.23.1"
transformers = "4.31.0"
einops = "0.6.1"
accelerate = "0.21.0"
bitsandbytes = "0.40.2"
scipy = "1.11.1"
pyngrok = "6.0.0"
pandas = "2.0.3"
datasets = "2.13.1"
evaluate = "0.4.0"
peft = "0.4.0"
trl = "0.4.7"
openai = "0.27.8"
passlib = "1.7.4"
bcrypt = "4.0.1"
langchain = "0.0.220"
scikit-learn = "1.3.0"
chromadb = "0.3.26"
pdfminer-six = "20221105"
docx2txt = "0.8"
python-multipart = "0.0.6"
tiktoken = "0.4.0"
pyepsilla = ">=0.1.1"
torch = "2.0.1"
posthog = "3.0.1"
pynvml = "11.5.0"
[tool.poetry.group.dev.dependencies]
pre-commit = "^3.3.3"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"