summaryrefslogtreecommitdiff
path: root/pyproject.toml
blob: bbf12882bec8d114d8579d0e43c07d56bf89e6a7 (plain)
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
[project]
name = "gemma3_4b"
version = "0.1.0"
description = "Add your description here"
requires-python = ">=3.13"
dependencies = [
    "absl-py>=2.3.1",
    "accelerate>=1.11.0",
    "bitsandbytes>=0.48.2",
    "datasets>=4.4.1",
    "diskcache>=5.6.3",
    "huggingface-hub>=0.36.0",
    "ipdb>=0.13.13",
    "ninja>=1.13.0",
    "openai>=2.8.0",
    "packaging>=25.0",
    "peft>=0.18.0",
    "pillow>=12.0.0",
    "sacrebleu>=2.5.1",
    "termcolor>=3.2.0",
    "torchvision>=0.24.1",
    "transformers>=4.57.1",
    "trl>=0.25.1",
    "unbabel-comet>=2.2.7",
]

[tool.uv.sources]
trl = { git = "https://github.com/huggingface/trl.git" }