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
|
[project]
name = "CaptionTranslation"
version = "0.1.0"
description = "Finetuning gemma3 towards translating captions"
requires-python = ">=3.13"
dependencies = [
"absl-py>=2.3.1",
"accelerate>=1.11.0",
"adamw-bf16>=0.0.3",
"bitsandbytes>=0.48.2",
"datasets>=4.4.1",
"diskcache>=5.6.3",
"hf_transfer",
"huggingface-hub>=0.36.0",
"ipdb>=0.13.13",
"jsonargparse[signatures]>=4.35.0",
"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",
"torchao>=0.14.1",
"torchvision>=0.24.1",
"transformers>=4.57.1",
"trl>=0.25.1",
]
[tool.uv.sources]
trl = { git = "https://github.com/huggingface/trl.git" }
|
