-
Notifications
You must be signed in to change notification settings - Fork 228
/
pyproject.toml
84 lines (75 loc) · 2.27 KB
/
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"
[project]
name = "prismatic"
authors = [
{name = "Siddharth Karamcheti", email="[email protected]"},
{name = "Suraj Nair", email="[email protected]"},
{name = "Ashwin Balakrishna", email="[email protected]"},
]
description = "Prismatic VLMs: Investigating the Design Space of Visually-Conditioned Language Models"
version = "0.0.2"
readme = "README.md"
requires-python = ">=3.8"
keywords = ["vision-language models", "multimodal pretraining", "machine learning"]
license = {file = "LICENSE"}
classifiers = [
"Development Status :: 3 - Alpha",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3 :: Only",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
"accelerate>=0.25.0",
"draccus @ git+https://github.com/dlwh/draccus",
"einops",
# "flash_attn>=2.5.5", # Here for documentation -- install *AFTER* editable install (follow README)
"huggingface_hub",
"jsonlines",
"rich",
"sentencepiece",
"timm==0.9.10",
"torch>=2.1.0",
"torchvision>=0.16.0",
"torchaudio",
"transformers>=4.38.1",
"wandb"
]
[project.optional-dependencies]
dev = [
"black>=24.2.0",
"gpustat",
"ipython",
"pre-commit",
"ruff>=0.2.2",
]
[project.urls]
homepage = "https://github.com/TRI-ML/prismatic-vlms"
repository = "https://github.com/TRI-ML/prismatic-vlms"
documentation = "https://github.com/TRI-ML/prismatic-vlms"
[tool.setuptools.packages.find]
where = ["."]
exclude = ["cache"]
[tool.setuptools.package-data]
"prismatic" = ["py.typed"]
[tool.black]
line-length = 121
target-version = ["py38", "py39", "py310"]
preview = true
[tool.ruff]
line-length = 121
target-version = "py38"
[tool.ruff.lint]
select = ["A", "B", "E", "F", "I", "RUF", "W"]
ignore = ["F722"]
[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["E402", "F401"]