From 8363495f0a4c200318f56b35e4e01bbfc4bdc635 Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Mon, 27 Jun 2022 17:26:05 -0400 Subject: [PATCH] updated readme --- README.md | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 5b4df29..5224b36 100644 --- a/README.md +++ b/README.md @@ -1,10 +1,16 @@ # min(DALL·E) +[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/kuprel/min-dalle/blob/main/min(DALL·E).ipynb) + This is a minimal implementation of [DALL·E Mini](https://github.com/borisdayma/dalle-mini) in both Flax and PyTorch ### Setup -Run `sh setup.sh` to install dependencies and download pretrained models. The only required dependencies are `flax` and `torch`. In the bash script, GitHub LFS is used to download the VQGan detokenizer and the Weight & Biases python package is used to download the DALL·E Mini and DALL·E Mega transformer models. Those files can also be downloaded manually by visting the links in the bash script. +Run `sh setup.sh` to install dependencies and download pretrained models. The only required dependencies are `flax` and `torch`. In the bash script, GitHub LFS is used to download the VQGan detokenizer and the Weight & Biases python package is used to download the DALL·E Mini and DALL·E Mega transformer models. These models can also be downloaded manually: + +VQGan: https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384 +DALL·E Mini: https://wandb.ai/dalle-mini/dalle-mini/artifacts/DalleBart_model/mini-1/v0/files +DALL·E Mega: https://wandb.ai/dalle-mini/dalle-mini/artifacts/DalleBart_model/mega-1-fp16/v14/files ### Run