FlagEmbedding/setup.py

15 lines
368 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',
2023-08-04 12:44:30 +08:00
package_dir={"": "flag_embedding/baai_general_embedding"},
packages=find_packages("flag_embedding/baai_general_embedding"),
2023-08-02 17:40:00 +08:00
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
],
)