-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpyproject.toml
More file actions
32 lines (28 loc) · 790 Bytes
/
pyproject.toml
File metadata and controls
32 lines (28 loc) · 790 Bytes
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
[tool.poetry]
name = "PixelAI"
version = "0.1.0"
description = ""
authors = ["Andreas Doering <[email protected]>"]
readme = "README.md"
packages = [{include = "pixelai"}]
[tool.poetry.dependencies]
python = ">=3.12,<4.0"
numpy = "^2.1.2"
torch = {version = "^2.6.0", source = "pytorch-cu124"}
torchvision = {version = "^0.21.0", source = "pytorch-cu124"}
matplotlib = "^3.10.3"
scikit-learn = "^1.7.0"
diffusers = "^0.34.0"
accelerate = "^1.8.1"
tensorboard = "^2.19.0"
pytest = "^8.4.1"
[build-system]
requires = ["poetry-core>=2.0.0,<3.0.0"]
build-backend = "poetry.core.masonry.api"
[[tool.poetry.source]]
name = "PyPI"
priority = "primary"
[[tool.poetry.source]]
name = "pytorch-cu124"
url = "https://download.pytorch.org/whl/cu124"
priority = "supplemental"