mirror of
https://github.com/FlagOpen/FlagEmbedding.git
synced 2025-06-27 02:39:58 +00:00
22 lines
578 B
Python
22 lines
578 B
Python
from setuptools import setup, find_packages
|
|
|
|
with open("README.md", mode="r", encoding="utf-8") as readme_file:
|
|
readme = readme_file.read()
|
|
|
|
setup(
|
|
name='LM_Cocktail',
|
|
version='0.0.5',
|
|
description='LM_Cocktail',
|
|
long_description=readme,
|
|
long_description_content_type="text/markdown",
|
|
author_email='2906698981@qq.com',
|
|
url='https://github.com/FlagOpen/FlagEmbedding/LM_Cocktail',
|
|
packages=find_packages(),
|
|
install_requires=[
|
|
'torch>=1.6.0',
|
|
'transformers>=4.18.0',
|
|
'datasets',
|
|
'accelerate>=0.20.1'
|
|
],
|
|
)
|