mirror of
https://github.com/bigscience-workshop/petals
synced 2024-11-10 01:10:33 +00:00
1a78638c02
We avoid importing bitsandbytes when it's not used, since bitsandbytes doesn't always find correct CUDA libs and may raise exceptions because of that.
63 lines
1.8 KiB
INI
63 lines
1.8 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.40.0.post4
|
|
accelerate>=0.16.0,<0.21.0
|
|
huggingface-hub>=0.11.1,<1.0.0
|
|
tokenizers>=0.13.3
|
|
transformers>=4.30.1,<5.0.0
|
|
speedtest-cli==2.1.3
|
|
pydantic>=1.8.1,<2.0 # 2.0 is incompatible with hivemind==1.1.8
|
|
hivemind==1.1.8
|
|
tensor_parallel==1.0.23
|
|
humanfriendly
|
|
async-timeout>=4.0.2
|
|
cpufeature>=0.2.0
|
|
packaging>=20.9
|
|
sentencepiece>=0.1.99
|
|
peft@git+https://github.com/huggingface/peft@5884bdbea49e5e71e2cd06ecfa484bb635063735
|
|
safetensors>=0.3.1
|
|
|
|
[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
|