-
Notifications
You must be signed in to change notification settings - Fork 940
/
setup.py
36 lines (30 loc) · 1.26 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
"""
Copyright (c) 2022, salesforce.com, inc.
All rights reserved.
SPDX-License-Identifier: BSD-3-Clause
For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
"""
from setuptools import setup, find_namespace_packages
import platform
DEPENDENCY_LINKS = []
if platform.system() == "Windows":
DEPENDENCY_LINKS.append("https://download.pytorch.org/whl/torch_stable.html")
def fetch_requirements(filename):
with open(filename) as f:
return [ln.strip() for ln in f.read().split("\n")]
setup(
name="salesforce-lavis",
version="1.0.1",
author="Dongxu Li, Junnan Li, Hung Le, Guangsen Wang, Silvio Savarese, Steven C.H. Hoi",
description="LAVIS - A One-stop Library for Language-Vision Intelligence",
long_description=open("README.md", "r", encoding="utf-8").read(),
long_description_content_type="text/markdown",
keywords="Vision-Language, Multimodal, Image Captioning, Generative AI, Deep Learning, Library, PyTorch",
license="3-Clause BSD",
packages=find_namespace_packages(include="lavis.*"),
install_requires=fetch_requirements("requirements.txt"),
python_requires=">=3.7.0",
include_package_data=True,
dependency_links=DEPENDENCY_LINKS,
zip_safe=False,
)