pip and conda cpu install

README.md

@@ -12,4 +12,10 @@
 * integrate CLIP for better results (only if we have the time)
 * work on a demo (streamlit or colab or maybe just HF widget)
 * document (set up repo on model hub per instructions, start on README writeup…)
-* help with coordinating activities & progress
+* help with coordinating activities & progress
+
+
+## Dependencies Installation
+You should create a new python virtual environment and install the project dependencies inside the virtual env: `pip install -r requirements.txt`
+
+If you use `conda`, you can create the virtual env and install everything using: `conda env update -f environments.yaml`

environment.yaml

@@ -0,0 +1,10 @@
+name: dalle
+channels:
+  - defaults
+dependencies:
+  - python=3.9.5
+  - pip=21.1.3
+  - ipython=7.22.0
+  - cudatoolkit
+  - pip:
+    - -r requirements.txt

requirements.txt

@@ -0,0 +1,5 @@
+jax
+flax
+optax
+datasets
+git+https://github.com/huggingface/transformers