causal_pool_training/pyproject.toml at main · ScottCTD/causal_pool_training · GitHub

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
32
33
34
35
36
37
38
39
[project]
name = "causal-pool"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "bitsandbytes>=0.48.2",
    "datasets>=4.4.1",
    "huggingface-hub>=0.36.0",
    "ipywidgets>=8.1.8",
    "jsonlines>=4.0.0",
    "notebook>=7.4.7",
    "numpy>=2.3.4",
    "openai>=2.8.0",
    "qwen-vl-utils>=0.0.14",
    "torch==2.8.0",
    "torchaudio==2.8.0",
    "torchvision==0.23.0",
    "tenacity>=9.0.0",
    "tqdm>=4.67.1",
    "transformers>=4.57.1",
    "trl[peft]>=0.25.1",
    "wandb>=0.23.0",
    "flash-attn",
    "matplotlib>=3.10.7",
    "nvitop>=1.6.0",
    "google-genai>=1.52.0",
    "hydra-core>=1.3.2",
]

[project.optional-dependencies]
dev = [
    "pytest>=8.0.0",
    "pytest-asyncio>=0.23.0",
]

[tool.uv.sources]
flash-attn = { url = "https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiTRUE-cp312-cp312-linux_x86_64.whl" }