-
Notifications
You must be signed in to change notification settings - Fork 42
/
pyproject.toml
30 lines (27 loc) · 697 Bytes
/
pyproject.toml
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
[build-system]
requires = ["setuptools>=61"]
build-backend = "setuptools.build_meta"
[project]
name = "aim"
version = "1.0.0"
description = "Scalable Pre-training of Large Autoregressive Vision Models"
requires-python = ">=3.8"
readme = {file = "README.md", content-type = "text/markdown"}
license = {file = "LICENSE"}
authors = [
{name = "Apple"}
]
dependencies = [
"torch>=2.0.0",
"torchvision",
"huggingface_hub",
]
[project.optional-dependencies]
test = [
"pytest",
"numpy",
]
[tool.isort]
profile = "black"
sections = ["FUTURE", "STDLIB", "THIRDPARTY", "NUMERIC", "FIRSTPARTY", "LOCALFOLDER"]
known_numeric = ["torch", "torchvision", "numpy", "jax", "flax", "mlx"]