mirror of
https://github.com/bigscience-workshop/petals
synced 2024-10-31 09:20:41 +00:00
65 lines
1.9 KiB
INI
65 lines
1.9 KiB
INI
[metadata]
|
|
name = petals
|
|
version = attr: petals.__version__
|
|
author = Petals Developers
|
|
author_email = petals-devs@googlegroups.com
|
|
description = Easy way to efficiently run 100B+ language models without high-end GPUs
|
|
long_description = file: README.md
|
|
long_description_content_type = text/markdown
|
|
url = https://github.com/bigscience-workshop/petals
|
|
project_urls =
|
|
Bug Tracker = https://github.com/bigscience-workshop/petals/issues
|
|
classifiers =
|
|
Development Status :: 4 - Beta
|
|
Intended Audience :: Developers
|
|
Intended Audience :: Science/Research
|
|
License :: OSI Approved :: MIT License
|
|
Programming Language :: Python :: 3
|
|
Programming Language :: Python :: 3.8
|
|
Programming Language :: Python :: 3.9
|
|
Programming Language :: Python :: 3.10
|
|
Programming Language :: Python :: 3.11
|
|
Topic :: Scientific/Engineering
|
|
Topic :: Scientific/Engineering :: Mathematics
|
|
Topic :: Scientific/Engineering :: Artificial Intelligence
|
|
Topic :: Software Development
|
|
Topic :: Software Development :: Libraries
|
|
Topic :: Software Development :: Libraries :: Python Modules
|
|
|
|
[options]
|
|
package_dir =
|
|
= src
|
|
packages = find:
|
|
python_requires = >=3.8
|
|
install_requires =
|
|
torch>=1.12
|
|
bitsandbytes==0.41.1
|
|
accelerate>=0.22.0
|
|
huggingface-hub>=0.11.1,<1.0.0
|
|
tokenizers>=0.13.3
|
|
transformers>=4.32.0,<5.0.0 # if you change this, please also change version assert in petals/__init__.py
|
|
speedtest-cli==2.1.3
|
|
pydantic>=1.10,<2.0 # 2.0 is incompatible with hivemind yet
|
|
hivemind==1.1.10.post2
|
|
tensor_parallel==1.0.23
|
|
humanfriendly
|
|
async-timeout>=4.0.2
|
|
cpufeature>=0.2.0; platform_machine == "x86_64"
|
|
packaging>=20.9
|
|
sentencepiece>=0.1.99
|
|
peft>=0.5.0
|
|
safetensors>=0.3.1
|
|
Dijkstar>=2.6.0
|
|
|
|
[options.extras_require]
|
|
dev =
|
|
pytest==6.2.5
|
|
pytest-forked
|
|
pytest-asyncio==0.16.0
|
|
black==22.3.0
|
|
isort==5.10.1
|
|
psutil
|
|
|
|
[options.packages.find]
|
|
where = src
|