pyproject.toml 552 B

123456789101112131415161718192021222324252627282930
  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. "accelerate>=0.23.0",
  15. "bitsandbytes>=0.41.1",
  16. "peft>=0.5.0",
  17. "omegaconf>=2.3.0",
  18. ]
  19. requires-python = ">=3.10"
  20. license = {text = "MIT"}
  21. [tool.setuptools]
  22. py-modules = ["speech_lm"]
  23. [tool.pdm.dev-dependencies]
  24. dev = [
  25. "black>=23.9.1",
  26. "isort>=5.12.0",
  27. ]