diff --git a/README.md b/README.md index e08eaa2..8e83657 100644 --- a/README.md +++ b/README.md @@ -1067,26 +1067,12 @@ dataset = ImageEmbeddingDataset( ) ``` -### Scripts (wip) +### Scripts #### `train_diffusion_prior.py` For detailed information on training the diffusion prior, please refer to the [dedicated readme](prior.md) -## CLI (wip) - -```bash -$ dream 'sharing a sunset at the summit of mount everest with my dog' -``` - -Once built, images will be saved to the same directory the command is invoked - -template - -## Training CLI (wip) - -template - ## Todo - [x] finish off gaussian diffusion class for latent embedding - allow for prediction of epsilon