setup.cfg 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  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.37.1
  35. accelerate==0.15.0
  36. huggingface-hub==0.11.1
  37. transformers>=4.25.1,<5.0.0
  38. speedtest-cli==2.1.3
  39. hivemind @ git+https://github.com/learning-at-home/hivemind.git
  40. tensor_parallel==1.0.23
  41. humanfriendly
  42. async-timeout>=4.0.2
  43. packaging>=20.9
  44. [options.extras_require]
  45. dev =
  46. pytest==6.2.5
  47. pytest-forked
  48. pytest-asyncio==0.16.0
  49. black==22.3.0
  50. isort==5.10.1
  51. psutil
  52. [options.packages.find]
  53. where = src