[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" }