mechinterp-modular-arithmetic/pyproject.toml at main · randommm/mechinterp-modular-arithmetic · 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
40
41
42
43
44
45
[project]
name = "mechinterp-modular-arithmetic"
version = "0.1.0"
readme = "README.md"
requires-python = ">=3.13,<3.14"
dependencies = [
    "accelerate>=1.11.0",
    "compressed-tensors>=0.17.0",
    "datasets>=4.8.4",
    "ipdb>=0.13",
    "ipython>=9",
    "jupyter>=1.1.1",
    "kernels<0.15",
    "matplotlib>=3.10.9",
    "numpy>=2.4.4",
    "peft>=0.17.1",
    "pyqt6>=6.11.0",
    "trl>=0.23.1",
    "tiktoken>=0.12.0",
    "torch==2.11.0+rocm7.2",
    "torchvision>=0.22.0",
    "transformers>=4.57.1",
    "transformer-lens>=3.3.0",
    "triton-rocm",
    "zarr>=3.1.0,<3.2",
    "polars>=1.41.2",
    "pyvene>=0.1.8",
]

[tool.uv]
environments = ["sys_platform == 'linux'"]
override-dependencies = [
    "torchvision>=0.22.0",
]

[[tool.uv.index]]
name = "pytorch-rocm"
url = "https://download.pytorch.org/whl/rocm7.2"
explicit = true

[tool.uv.sources]
torch = { index = "pytorch-rocm" }
torchvision = { index = "pytorch-rocm" }
transformer-lens = { git = "https://github.com/randommm/TransformerLens.git", branch = "support-new" }
triton-rocm = { index = "pytorch-rocm" }