PAMCL/pixi.toml at main · sun2ot/PAMCL · 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
[workspace]
channels = ["conda-forge"]
name = "selfrec"
platforms = ["linux-64"]
version = "0.1.0"

[system-requirements]
cuda = "11"

[environments]
default = {features = ["default"], solve-group = "prod"}
dev = {features = ["default", "dev"], solve-group = "prod"}

[dependencies]
python = "3.10.*"
scipy = ">=1.13.0,<1.14"
safetensors = ">=0.4.0,<0.5"
numba = ">=0.60.0,<0.61"
numpy = ">=1.26.0,<1.27"
pyyaml = ">=6.0,<7"
pandas = ">=2.2.0,<2.3"
scikit-learn = ">=1.5.0,<1.6"
pillow = ">=10.4.0,<10.5"
matplotlib = ">=3.9.0,<4.0"
ipykernel = ">=6.28.0,<7"
python-dotenv = ">=0.21.0,<0.22"
tqdm = ">=4.67.1,<5"

[pypi-options]
no-build-isolation = ["flash-attn"]

[pypi-dependencies]
torch = { url = "https://mirrors.aliyun.com/pytorch-wheels/cu118/torch-2.1.0+cu118-cp310-cp310-linux_x86_64.whl"}
qywx-bot = "==1.4"
requests = ">=2.31.0"
psutil = ">=7.1.3"

[feature.dev.pypi-dependencies]
flash-attn = "==2.6.3"
# jq = "==1.8.0"
transformers = ">=4.44.0,<4.45"
sentence-transformers = "==3.1.1"