boris commited on
Commit
5da4af0
1 Parent(s): ab5769a

doc: add reference to inference pipeline in README

Browse files
Files changed (1) hide show
  1. README.md +9 -7
README.md CHANGED
@@ -22,10 +22,6 @@ You can create your own pictures with [the demo](https://huggingface.co/spaces/f
22
 
23
  Refer to [our report](https://wandb.ai/dalle-mini/dalle-mini/reports/DALL-E-mini--Vmlldzo4NjIxODA).
24
 
25
- ## Where does the logo come from?
26
-
27
- The "armchair in the shape of an avocado" was used by OpenAI when releasing DALL·E to illustrate the model's capabilities. Having successful predictions on this prompt represents a big milestone to us.
28
-
29
  ## Development
30
 
31
  This section is for the adventurous people wanting to look into the code.
@@ -58,13 +54,19 @@ Use [patil-suraj/vqgan-jax](https://github.com/patil-suraj/vqgan-jax).
58
 
59
  ### Training of Seq2Seq
60
 
61
- Refer to `dev/seq2seq` folder.
62
 
63
  You can also adjust the [sweep configuration file](https://docs.wandb.ai/guides/sweeps) if you need to perform a hyperparameter search.
64
 
65
- ### Inference
 
 
66
 
67
- Refer to `dev/notebooks/demo`.
 
 
 
 
68
 
69
  ## Authors
70
 
 
22
 
23
  Refer to [our report](https://wandb.ai/dalle-mini/dalle-mini/reports/DALL-E-mini--Vmlldzo4NjIxODA).
24
 
 
 
 
 
25
  ## Development
26
 
27
  This section is for the adventurous people wanting to look into the code.
 
54
 
55
  ### Training of Seq2Seq
56
 
57
+ Refer to [`dev/seq2seq`](dev/seq2seq) folder.
58
 
59
  You can also adjust the [sweep configuration file](https://docs.wandb.ai/guides/sweeps) if you need to perform a hyperparameter search.
60
 
61
+ ### Inference Pipeline
62
+
63
+ To generate sample predictions and understand the inference pipeline step by step, refer to [`dev/inference/inference_pipeline.ipynb`](dev/inference/inference_pipeline.ipynb).
64
 
65
+ [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/borisdayma/dalle-mini/blob/main/dev/inference/inference_pipeline.ipynb)
66
+
67
+ ## Where does the logo come from?
68
+
69
+ The "armchair in the shape of an avocado" was used by OpenAI when releasing DALL·E to illustrate the model's capabilities. Having successful predictions on this prompt represents a big milestone to us.
70
 
71
  ## Authors
72