mirror of
https://github.com/bigscience-workshop/petals
synced 2024-10-31 09:20:41 +00:00
0a313bf6c5
This PR: 1. Updates hivemind to 1.1.8 (includes https://github.com/learning-at-home/hivemind/pull/565) 2. Enables efficient bfloat16 serialization by default (`USE_LEGACY_BFLOAT16 = False`) 3. Removes logging code that was included to hivemind in https://github.com/learning-at-home/hivemind/pull/542
58 lines
1.6 KiB
INI
58 lines
1.6 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.7
|
|
Programming Language :: Python :: 3.8
|
|
Programming Language :: Python :: 3.9
|
|
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.7
|
|
install_requires =
|
|
torch>=1.12
|
|
bitsandbytes==0.38.0.post2
|
|
accelerate>=0.16.0,<1.0.0
|
|
huggingface-hub>=0.11.1,<1.0.0
|
|
transformers>=4.25.1,<5.0.0
|
|
speedtest-cli==2.1.3
|
|
hivemind==1.1.8
|
|
tensor_parallel==1.0.23
|
|
humanfriendly
|
|
async-timeout>=4.0.2
|
|
cpufeature>=0.2.0
|
|
packaging>=20.9
|
|
|
|
[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
|