forked from TomFrederik/unseal
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
49 lines (46 loc) · 1.21 KB
/
setup.py
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
from setuptools import setup, find_packages
import os, sys
sys.path.insert(0, os.path.abspath("."))
from unseal import __version__
version = __version__
with open("README.md", "r") as fh:
long_description = fh.read()
setup(
name='Unseal',
version=version,
packages=find_packages(exclude=[]),
python_requires='>=3.6.0',
install_requires=[
'torch',
'einops>=0.3.2',
'numpy',
'transformers',
'tqdm',
'matplotlib',
'streamlit',
],
# entry_points={
# 'console_scripts': [
# '"unseal compare" = unseal.commands.interfaces.compare_two_inputs:main',
# ]
# },
description=(
"Unseal: "
"A collection of infrastructure and tools for research in "
"transformer interpretability."
),
long_description=long_description,
long_description_content_type="text/markdown",
author="The Unseal Team",
author_email="[email protected]",
url="https://github.com/TomFrederik/unseal/",
license="Apache License 2.0",
keywords=[
"pytorch",
"tensor",
"machine learning",
"neural networks",
"interpretability",
"transformers",
],
)