--- a +++ b/setup.py @@ -0,0 +1,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, +)