[project]
name = "TADBot"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "ruff>=0.7.3",
    "torch==2.5.1+cu124",
    "torchvision==0.20.1+cu124",
    "torchaudio==2.5.1+cu124",
    "huggingface-hub>=0.26.2",
    "timm>=1.0.11",
    "thop>=0.1.1.post2209072238",
    "deepface==0.0.93",
    "bitsandbytes>=0.44.1",
    "peft>=0.13.2",
    "trl>=0.12.1",
    "ninja==1.11.1.1",
    "numpy>=1.26.4",
    "transformers>=4.46.2",
    "datasets>=3.1.0",
    "tensorboard>=2.18.0",
    "jsonlines>=4.0.0",
    "python-dotenv>=1.0.1",
    "ipykernel>=6.29.5",
    "ipywidgets>=8.1.5",
    "pyyaml>=6.0.2",
    "torch-tb-profiler>=0.4.3",
    "tensorflow>=2.18.0",
    "wandb>=0.18.7",
]

[tool.uv.sources]
torch = { index = "pytorch" }
torchvision = { index = "pytorch" }
torchaudio = { index = "pytorch" }
deepface = {git = "https://github.com/serengil/deepface", tag="v0.0.93"}

[[tool.uv.index]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu124"
explicit = true

[dependency-groups]
dev = [
    "deepface>=0.0.93",
    "numpy<2",
    "opencv-python>=4.10.0.84",
    "thop>=0.1.1.post2209072238",
    "timm>=1.0.11",
    "torch>=2.5.1",
    "torchaudio>=2.5.1",
    "torchvision>=0.20.1",
]