FlagEmbedding/setup.py

14 lines
303 B
Python
Raw Normal View History

2023-08-02 17:40:00 +08:00
from setuptools import setup, find_packages
setup(
name='FlagEmbedding',
version='0.0.1',
package_dir={"": "universal_embedding"},
packages=find_packages("universal_embedding"),
install_requires=[
'torch>=1.6.0',
'transformers>=4.18.0',
'datasets',
],
)