tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

separate setup processes for flax and torch

Browse Source
This commit is contained in:
Brett Kuprel
2022-07-01 11:08:33 -04:00
parent 7bf76deafb
commit 09a0f85b8e
5 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
Vendored
+2 -1
View File
@@ -9,7 +9,8 @@ It currently takes **7.4 seconds** to generate an image with DALL·E Mega with P
### Setup
Run `sh setup.sh` to install dependencies and download pretrained models. The models can also be downloaded manually here:
Run either `sh setup_torch.sh` or `sh setup_flax.sh` to install dependencies and download pretrained models. The torch models can be manually downloaded [here](https://huggingface.co/kuprel/min-dalle).
The flax models can be manually downloaded here:
[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)