requirements.txt 375 B

1234567891011121314
  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. torchmetrics==0.6.2
  4. bitsandbytes-cuda111==0.26.0
  5. transformers==4.12.2
  6. tokenizers==0.10.3
  7. datasets==1.14.0
  8. torch-optimizer==0.1.0
  9. wandb==0.12.1
  10. nltk==3.6.2
  11. sentencepiece==0.1.96
  12. aiohttp==3.7.4.post0
  13. requests>=2.23.0
  14. termcolor==1.1.0