-
Notifications
You must be signed in to change notification settings - Fork 29
/
pyproject.toml
39 lines (34 loc) · 1.12 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
[tool.poetry]
name = "minicons"
version = "0.2.48"
description = "A package of useful functions to analyze transformer based language models."
authors = ["Kanishka Misra <kanishka.replies@gmail.com>", "Aaron Mueller <amueller@jhu.edu>", "Carina Kauf <ckauf@mit.edu>", "Wentao Wang <wentao.wang@nyu.edu>", "Maria Ryskina <ryskina@mit.edu>"]
license = "MIT"
readme = "README.md"
homepage = "https://github.com/kanishkamisra/minicons"
repository = "https://github.com/kanishkamisra/minicons"
keywords = ["transformers", "language models", "nlp", "interpretability"]
[tool.poetry.scripts]
minicons = 'minicons.bin.score_cli:process'
[tool.poetry.dependencies]
python = ">=3.9.0,<4"
transformers = "^4.30.0"
accelerate= "^0.25.0"
torch = "^2.0.0"
urllib3 = "^1.26.7"
pandas = "2.2.0"
openai = "^0.28.0"
tenacity = "^8.2.3"
pillow = "^10.2.0"
wonderwords = "^2.2.0"
[tool.poetry.dev-dependencies]
jupyter = "^1.0.0"
Sphinx = "^4.3.1"
furo = "^2021.11.23"
myst-parser = "^0.16.0"
typed-ast = "^1.5.5"
[tool.poetry.group.dev.dependencies]
black = ">=23.1,<25.0"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"