setup.cfg 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  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.8
  19. Programming Language :: Python :: 3.9
  20. Programming Language :: Python :: 3.10
  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.8,<3.11
  32. install_requires =
  33. torch>=1.12
  34. bitsandbytes==0.40.1.post1
  35. accelerate>=0.20.3,<0.21.0
  36. huggingface-hub>=0.11.1,<1.0.0
  37. tokenizers>=0.13.3
  38. transformers>=4.31.0,<5.0.0
  39. speedtest-cli==2.1.3
  40. pydantic>=1.10,<2.0 # 2.0 is incompatible with hivemind==1.1.8
  41. hivemind==1.1.8
  42. tensor_parallel==1.0.23
  43. humanfriendly
  44. async-timeout>=4.0.2
  45. cpufeature>=0.2.0
  46. packaging>=20.9
  47. sentencepiece>=0.1.99
  48. peft>=0.4.0
  49. safetensors>=0.3.1
  50. Dijkstar>=2.6.0
  51. [options.extras_require]
  52. dev =
  53. pytest==6.2.5
  54. pytest-forked
  55. pytest-asyncio==0.16.0
  56. black==22.3.0
  57. isort==5.10.1
  58. psutil
  59. [options.packages.find]
  60. where = src