textforge/pyproject.toml at main · ameen-91/textforge · GitHub

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 = "textforge"
version = "0.1.2"
description = "Automated Synthetic Data Generation, Distillation, Quantization and Deployment Pipeline."
license = "MIT"
readme = "README.md"
authors = ["ameen-91 <mohammedameen9011@gmail.com>"]

[tool.poetry.scripts]
textforge = "textforge.cli:main"

[tool.poetry.dependencies]
python = "^3.11"
openai = "^1.63.0"
pandas = "2.2.2"
tqdm = "^4.67.1"
transformers = "^4.48.3"
peft = "^0.14.0"
datasets = "^3.2.0"
evaluate = "^0.4.3"
scikit-learn = "^1.6.1"
onnxruntime = "^1.20.1"
typer = "^0.15.1"
onnx = "^1.17.0"
ipywidgets = "^8.1.5"
google-genai = "^1.2.0"
pytest = "^8.3.4"
python-dotenv = "^1.0.1"
seaborn = "^0.13.2"
fastapi = {extras = ["standard"], version = "^0.115.8"}
jinja2 = "^3.1.5"


[tool.poetry.group.dev.dependencies]
ipykernel = "^6.29.5"
mkdocstrings-python = "^1.16.2"
mkdocstrings = "^0.28.2"
mkdocs-material = {extras = ["imaging"], version = "^9.6.5"}
pytest = "^8.3.4"
pytest-asyncio = "^0.25.3"
pytest-cov = "^6.0.0"
coverage-badge = "^1.1.2"
seaborn = "^0.13.2"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"