-
Notifications
You must be signed in to change notification settings - Fork 415
/
setup.py
executable file
·45 lines (44 loc) · 1.35 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
# Copyright © 2022 BAAI. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License")
from setuptools import find_packages, setup
setup(
name="flagai",
version="v1.8.4",
description="FlagAI aims to help researchers and developers to freely train and test large-scale models for NLP/CV/VL tasks.",
long_description=open("README.md", encoding="utf-8").read(),
long_description_content_type="text/markdown",
author="FlagAI-Open",
author_email="open@baai.ac.cn",
url="https://github.com/FlagAI-Open/FlagAI",
packages=find_packages(exclude="tests"), # same as name
license="Apache 2.0",
include_package_data=True,
python_requires=">=3.8",
install_requires=[
'nltk>=3.6.7',
'sentencepiece>=0.1.96',
'boto3>=1.17.32',
'pandas>=1.3.5',
'jieba>=0.42.1',
'scikit-learn>=1.0.2',
'tensorboard>=2.9.0',
'transformers>=4.31.0',
'datasets>=2.0.0',
'setuptools>=66.0.0',
'protobuf>=3.19.6',
'ftfy',
'Pillow>=9.3.0',
'einops>=0.3.0',
'diffusers>=0.7.2',
'pytorch-lightning>=1.6.5',
'taming-transformers-rom1504==0.0.6',
'rouge-score',
'sacrebleu>=2.3.1',
'jsonlines',
'accelerate',
'PyYAML>=5.4.1',
'safetensors',
'timm',
]
)