requirements.txt 308 B

12345678910111213
  1. git+git://github.com/learning-at-home/hivemind@dalle-v1
  2. git+git://github.com/learning-at-home/dalle-pytorch@weight-sharing
  3. bitsandbytes-cuda111>=0.26.0
  4. transformers==4.12.2
  5. tokenizers>=0.10.2
  6. datasets>=1.11.0
  7. torch_optimizer>=0.1.0
  8. wandb>=0.10.33
  9. nltk>=3.6.2
  10. sentencepiece
  11. aiohttp
  12. requests>=2.24.0
  13. termcolor