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

added to pypi

Browse Source
This commit is contained in:
Brett Kuprel
2022-07-01 18:16:55 -04:00
parent f0c8f258e9
commit be2beca7c0
9 changed files with 116 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
Vendored
+4 -2
View File
@@ -9,14 +9,16 @@ It currently takes **7.4 seconds** to generate an image with DALL·E Mega with P
The flax model, and the code for coverting it to torch, have been moved [here](https://github.com/kuprel/min-dalle-flax).
### Setup
### Install
Run `sh setup.sh` to install dependencies and download pretrained models. The torch models can be manually downloaded [here](https://huggingface.co/kuprel/min-dalle/tree/main).
```$ pip install min-dalle```
### Usage
Use the python script `image_from_text.py` to generate images from the command line. Note: the command line script loads the models and parameters each time. To load a model once and generate multiple times, initialize `MinDalleTorch`, then call `generate_image` with some text and a seed. See the colab for an example.
Model parameters will be downloaded as needed to the directory specified. The models can also be manually downloaded [here](https://huggingface.co/kuprel/min-dalle/tree/main).
### Examples
```