diff --git a/README.md b/README.md
index 0a6f442..10056c5 100644
--- a/README.md
+++ b/README.md
@@ -31,6 +31,21 @@ As of 5/23/22, it is no longer SOTA. SOTA will be In-progress test run 🚧
- DALL-E 2 🚧
+## Appreciation
+
+This library would not have gotten to this working state without the help of
+
+- Zion for the distributed training code for the diffusion prior
+- Aidan for the distributed training code for the decoder as well as the dataloaders
+- Kumar for working on the initial diffusion training script
+- Romain for the pull request reviews and project management
+- He Cao and xiankgx for the Q&A and for identifying of critical bugs
+- Katherine for her advice
+- Stability AI for the generous sponsorship
+- 🤗 Huggingface and in particular Sylvain for the Accelerate library
+
+... and many others. Thank you! 🙏
+
## Install
```bash
@@ -1041,19 +1056,6 @@ Once built, images will be saved to the same directory the command is invoked
template
-## Appreciation
-
-This library would not have gotten to this working state without the help of
-
-- Zion and Kumar for the diffusion training script
-- Aidan for the decoder training script and dataloaders
-- Romain for the pull request reviews and project management
-- He Cao and xiankgx for the Q&A and for identifying of critical bugs
-- Katherine for her advice
-- Stability AI for the generous sponsorship
-
-... and many others. Thank you! 🙏
-
## Todo
- [x] finish off gaussian diffusion class for latent embedding - allow for prediction of epsilon
diff --git a/setup.py b/setup.py
index ab2ba08..411570a 100644
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,7 @@ setup(
'text to image'
],
install_requires=[
+ 'accelerate',
'click',
'clip-anytorch',
'coca-pytorch>=0.0.5',