pyproject.toml 608 B

1234567891011121314151617181920212223242526272829303132
  1. [project]
  2. name = "speech-lm"
  3. version = "0.0.1"
  4. description = ""
  5. authors = [
  6. {name = "Lengyue", email = "lengyue@lengyue.me"},
  7. ]
  8. dependencies = [
  9. "torch>=2.1.0",
  10. "torchvision>=0.16.0",
  11. "torchaudio>=2.1.0",
  12. "transformers>=4.34.0",
  13. "datasets>=2.14.5",
  14. "bitsandbytes>=0.41.1",
  15. "peft>=0.5.0",
  16. "deepspeed>=0.11.1",
  17. "lightning>=2.0.9.post0",
  18. "hydra-core>=1.3.2",
  19. "pyrootutils>=1.0.4",
  20. ]
  21. requires-python = ">=3.10"
  22. license = {text = "MIT"}
  23. [tool.setuptools]
  24. py-modules = ["speech_lm"]
  25. [tool.pdm.dev-dependencies]
  26. dev = [
  27. "black>=23.9.1",
  28. "isort>=5.12.0",
  29. ]