setup.cfg 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. [metadata]
  2. name = petals
  3. version = attr: petals.__version__
  4. author = Petals Developers
  5. author_email = petals-devs@googlegroups.com
  6. description = Easy way to efficiently run 100B+ language models without high-end GPUs
  7. long_description = file: README.md
  8. long_description_content_type = text/markdown
  9. url = https://github.com/bigscience-workshop/petals
  10. project_urls =
  11. Bug Tracker = https://github.com/bigscience-workshop/petals/issues
  12. classifiers =
  13. Development Status :: 4 - Beta
  14. Intended Audience :: Developers
  15. Intended Audience :: Science/Research
  16. License :: OSI Approved :: MIT License
  17. Programming Language :: Python :: 3
  18. Programming Language :: Python :: 3.7
  19. Programming Language :: Python :: 3.8
  20. Programming Language :: Python :: 3.9
  21. Topic :: Scientific/Engineering
  22. Topic :: Scientific/Engineering :: Mathematics
  23. Topic :: Scientific/Engineering :: Artificial Intelligence
  24. Topic :: Software Development
  25. Topic :: Software Development :: Libraries
  26. Topic :: Software Development :: Libraries :: Python Modules
  27. [options]
  28. package_dir =
  29. = src
  30. packages = find:
  31. python_requires = >=3.7
  32. install_requires =
  33. torch>=1.12
  34. bitsandbytes==0.38.0.post2
  35. accelerate>=0.16.0,<1.0.0
  36. huggingface-hub>=0.11.1,<1.0.0
  37. tokenizers>=0.13.3
  38. transformers>=4.30.1,<5.0.0
  39. speedtest-cli==2.1.3
  40. hivemind==1.1.8
  41. tensor_parallel==1.0.23
  42. humanfriendly
  43. async-timeout>=4.0.2
  44. cpufeature>=0.2.0
  45. packaging>=20.9
  46. sentencepiece>=0.1.99
  47. [options.extras_require]
  48. dev =
  49. pytest==6.2.5
  50. pytest-forked
  51. pytest-asyncio==0.16.0
  52. black==22.3.0
  53. isort==5.10.1
  54. psutil
  55. [options.packages.find]
  56. where = src