requirements.txt 328 B

12345678910111213
  1. https://github.com/learning-at-home/hivemind/archive/dalle-v1.zip
  2. https://github.com/learning-at-home/dalle-pytorch/archive/weight-sharing.zip
  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