File size: 481 Bytes
3d5551b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from setuptools import setup, find_packages
setup(
name="tevatron",
version="0.0.1",
packages=find_packages("src"),
package_dir={"": "src"},
url="https://github.com/texttron/tevatron",
license="Apache 2.0",
author="Luyu Gao",
author_email="[email protected]",
description="Tevatron: A toolkit for learning and running deep dense retrieval models.",
python_requires=">=3.7",
install_requires=["transformers>=4.10.0", "datasets>=1.1.3"],
) |