ModelForge/pyproject.toml at main · ForgeOpus/ModelForge · 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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
[project]
name = "modelforge-finetuning"
version = "3.0.5"
description = "ModelForge: A no-code toolkit for fine-tuning HuggingFace models"
authors = [
    {name = "R3tr0 M1ll3r", email = "r3tr0.m1ll3r@gmail.com"},
]
readme = "README.md"
requires-python = ">=3.11,<3.12"
dependencies = [
    "accelerate==1.5.2",
    "datasets==3.5.0",
    "dotenv>=0.9.9",
    "fastapi==0.115.12",
    "huggingface-hub==0.30.2",
    "safetensors==0.5.3",
    "setuptools==78.1.0",
    "tensorboard==2.19.0",
    "tensorboard-data-server==0.7.2",
    "tokenizers==0.21.0",
    "tqdm==4.67.1",
    "transformers==4.48.3",
    "trl==0.16.0",
    "uvicorn",
    "platformdirs",
    "psutil",
    "pynvml",
    "peft",
    "python-multipart",
    "sqlalchemy>=2.0.44",
]

[project.optional-dependencies]
quantization = [
    "bitsandbytes==0.45.3",
]
cli = [
    "questionary>=2.0.1",
    "rich>=13.0.0",
    "ipywidgets>=8.0.0",
    "ipython>=8.0.0",
]
unsloth = [
    "unsloth",
    "bitsandbytes==0.45.3",
]

[tool.setuptools.packages.find]
where = ["."]
include = ["ModelForge*"]

[project.scripts]
modelforge = "ModelForge.cli:main"
modelforge-nb = "ModelForge.notebook_cli.wizard:main"

[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[tool.setuptools]
include-package-data = true

[tool.setuptools.package-data]
"ModelForge" = [
    "**/*.jsonl",
    "**/*.yaml",
    "**/*.yml",
    "**/*.json",
    # Include the entire Frontend/build directory and all its contents
    "Frontend/build/**",
]

[tool.setuptools.exclude-package-data]
"*" = [
    "database/*",
    "datasets/*",
    "model_checkpoints/*",
    "training_logs/*",
    "*__pycache__*",
    "*.pyc"
]

[project.urls]
Documentation = "https://modelforge-finetuning.readthedocs.io"
Repository = "https://github.com/forgeopus/modelforge"
Organization = "https://github.com/forgeopus"