Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
lucidrains
GitHub Repository: lucidrains/vit-pytorch
 Path: blob/main/pyproject.toml
649 views
1
[build-system]

2
requires = ["setuptools>=61", "wheel"]

3
build-backend = "setuptools.build_meta"

4


5
[project]

6
name = "vit-pytorch"

7
version = "1.17.7"

8
description = "Vision Transformer (ViT) - Pytorch"

9
readme = { file = "README.md", content-type = "text/markdown" }

10
license = { file = "LICENSE" }

11
authors = [

12
  { name = "Phil Wang", email = "[email protected]" },

13
]

14
requires-python = ">=3.8"

15
keywords = [

16
  "artificial intelligence",

17
  "attention mechanism",

18
  "image recognition",

19
]

20
classifiers = [

21
  "Development Status :: 4 - Beta",

22
  "Intended Audience :: Developers",

23
  "Topic :: Scientific/Engineering :: Artificial Intelligence",

24
  "License :: OSI Approved :: MIT License",

25
  "Programming Language :: Python :: 3",

26
  "Programming Language :: Python :: 3 :: Only",

27
  "Programming Language :: Python :: 3.8",

28
  "Programming Language :: Python :: 3.9",

29
  "Programming Language :: Python :: 3.10",

30
  "Programming Language :: Python :: 3.11",

31
  "Programming Language :: Python :: 3.12",

32
]

33
dependencies = [

34
  "einops>=0.7.0",

35
  "torch>=1.10",

36
  "torchvision",

37
]

38


39
[project.optional-dependencies]

40
test = [

41
  "pytest",

42
  "torch==2.4.0",

43
  "torchvision==0.19.0",

44
]

45


46
[project.urls]

47
Homepage = "https://github.com/lucidrains/vit-pytorch"

48
Repository = "https://github.com/lucidrains/vit-pytorch"

49


50
[tool.setuptools]

51
include-package-data = true

52


53
[tool.setuptools.packages.find]

54
include = ["vit_pytorch*"]

55
exclude = ["examples*", "tests*", "test*"]

56


57
[tool.pytest.ini_options]

58
testpaths = ["tests", "."]

59
python_files = ["test_*.py", "*_test.py"]

60
addopts = "-q"

61
filterwarnings = [

62
  "ignore::FutureWarning",

63
]

64


65