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

fixed an issue with argument parser

Browse Source
This commit is contained in:
Brett Kuprel
2022-06-27 16:49:42 -04:00
parent e7001f063c
commit a014dccc05
3 changed files with 105 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+5 -6
View File
@@ -4,27 +4,26 @@ This is a minimal implementation of [DALL·E Mini](https://github.com/borisdayma
### 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. You can also download those files 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. Those files can also be downloaded manually by visting the links in the bash script.
### Run
Here are some examples
```
python3 image_from_text.py --text='alien life' --seed=7
python image_from_text.py --seed=7 --text='alien life'
```
![Alien](examples/alien.png)
```
python3 image_from_text.py --mega --seed=4 \
--text='a comfy chair that looks like an avocado'
python image_from_text.py --mega --seed=4 --text='a comfy chair that looks like an avocado'
```
![Avocado Armchair](examples/avocado_armchair.png)
```
python3 image_from_text.py --mega --seed=100 \
--text='court sketch of godzilla on trial'
python image_from_text.py --mega --seed=100 --text='court sketch of godzilla on trial'
```
![Godzilla Trial](examples/godzilla_trial.png)