h2oai / wave-image-styling-playground

A interactive playground to style and edit images, generate art and have fun.
MIT License
13 stars 2 forks source link

Add DALL-E mini support for `Image Prompt` mode #20

Closed dilithjay closed 1 year ago

dilithjay commented 1 year ago

This PR enables the mini version of the DALL-E mini model available here. I will add the additional setting-up steps to the Wiki, once the PR is merged.

Instructions for setting up the files for DALL-E mini:

cd models
git clone https://huggingface.co/dalle-mini/dalle-mini
cd dalle-mini
git lfs pull

cd ..
git clone https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384
cd vqgan_imagenet_f16_16384
git lfs pull

The left side panel now looks as follows. The default option for Model is Stable Diffusion which includes the previously available radio buttons. image

In addition to the seed the following settings are supported: image