|
[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", |
|
] |
|
|