53 lines
2.2 KiB
Python
53 lines
2.2 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from setuptools import setup, find_packages
|
|
import os
|
|
|
|
core_req = ['ftfy', 'regex', 'tqdm', 'pytorch==1.7.1', 'torchvision']
|
|
extra_requires={'cuda': ['cudatoolkit==11.0']}
|
|
|
|
setup(
|
|
name=name,
|
|
version='0.1.0',
|
|
author="OpenAI",
|
|
author_email="dev@vctr.ai",
|
|
description="CLIP by OpenAI",
|
|
long_description=open("README.md", "r", encoding="utf-8").read(),
|
|
long_description_content_type="text/markdown",
|
|
keywords="vector, embeddings, machinelearning, ai, artificialintelligence, nlp, pytorch, nearestneighbors, search, analytics, clustering, dimensionalityreduction",
|
|
license="Apache",
|
|
packages=find_packages(exclude=["tests*"]),
|
|
python_requires=">=3",
|
|
install_requires=core_req,
|
|
extra_requires=extra_requires,
|
|
classifiers=[
|
|
"Development Status :: 5 - Production/Stable",
|
|
"Intended Audience :: Developers",
|
|
"Intended Audience :: Education",
|
|
"Intended Audience :: Science/Research",
|
|
"Intended Audience :: Information Technology",
|
|
"Intended Audience :: Financial and Insurance Industry",
|
|
"Intended Audience :: Healthcare Industry",
|
|
"Intended Audience :: Manufacturing",
|
|
"License :: OSI Approved :: Apache Software License",
|
|
"Operating System :: OS Independent",
|
|
"Programming Language :: Python",
|
|
"Programming Language :: Python :: 3",
|
|
"Programming Language :: Python :: 3.4",
|
|
"Programming Language :: Python :: 3.5",
|
|
"Programming Language :: Python :: 3.6",
|
|
"Programming Language :: Python :: 3.7",
|
|
"Programming Language :: Python :: Implementation :: PyPy",
|
|
"Topic :: Database",
|
|
"Topic :: Internet :: WWW/HTTP :: Indexing/Search",
|
|
"Topic :: Multimedia :: Sound/Audio :: Conversion",
|
|
"Topic :: Multimedia :: Video :: Conversion",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"Topic :: Scientific/Engineering :: Image Recognition",
|
|
"Topic :: Scientific/Engineering :: Information Analysis",
|
|
"Topic :: Scientific/Engineering :: Visualization",
|
|
"Topic :: Software Development :: Libraries :: Application Frameworks",
|
|
],
|
|
)
|