-
Notifications
You must be signed in to change notification settings - Fork 6
/
setup.py
40 lines (35 loc) · 929 Bytes
/
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
from setuptools import setup, find_packages
with open("README.md", "r", encoding="utf-8") as f:
long_description = f.read()
install_requires = [
"colorama",
"colorlog",
"datasets",
"jieba",
"pypinyin",
"pytorch_lightning>=2.0.3",
"scikit_learn",
"scipy",
"seaborn",
"sentencepiece",
"setuptools",
"spacy",
"gradio",
"torchmetrics==0.11.4",
"transformers>=4.30.2",
]
setup(
name="litie",
version="0.2.5",
description="Pytorch-lightning Code Blocks for Information Extraction",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/xusenlinzy/lit-ie",
author="xusenlin",
author_email="1659821119@qq.com",
ikeywords=["deep learning", "pytorch", "AI"],
python_requires=">=3.7",
setup_requires=[],
packages=find_packages(),
install_requires=install_requires,
)