FlagEmbedding/setup.py

15 lines
332 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',
2023-08-03 10:55:42 +08:00
'accelerate>=0.20.1'
2023-08-02 17:40:00 +08:00
],
)