| """ | |
| 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, | |
| ) | |