FlagEmbedding/setup.py

15 lines
322 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-03 20:38:51 +08:00
package_dir={"": "flag_embedding"},
packages=find_packages("flag_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
],
)