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

update readme

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-05 07:32:52 -04:00
parent fee7636ce3
commit 8239c8dba4
2 changed files with 27 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
README.md vendored
View File

@ -29,53 +29,46 @@ Load the model parameters once and reuse the model to generate multiple images.
```python ```python
from min_dalle import MinDalle from min_dalle import MinDalle
model = MinDalle(is_mega=True, models_root='./pretrained') model = MinDalle(
is_mega=True,
is_reusable=True,
models_root='./pretrained'
)
``` ```
The required models will be downloaded to `models_root` if they are not already there. Once everything has finished initializing, call `generate_image` with some text as many times as you want. The required models will be downloaded to `models_root` if they are not already there. Once everything has finished initializing, call `generate_image` with some text as many times as you want.
```python ```python
text = 'Dali painting of WALL·E' image = model.generate_image(
image = model.generate_image(text, grid_size=4) 'Dali painting of WALL·E',
seed=-1,
grid_size=4,
log2_supercondition_factor=3
)
display(image) display(image)
``` ```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/dali_walle.jpg" alt="min-dalle" width="300"/> <img src="https://github.com/kuprel/min-dalle/raw/main/examples/dali_walle.jpg" alt="min-dalle" width="300"/>
```python Use a positive `seed` for reproducible results. Higher values for `log2_supercondition_factor` result in better agreement with the text but a narrower variety of generated images.
text = 'Rusty Iron Man suit found abandoned in the woods being reclaimed by nature'
image = model.generate_image(text, grid_size=3) If the model is being used interactively (e.g. in a notebook) `generate_image_stream` can be used to generate a stream of images as it the model is decoding. The detokenizer adds a slight delay for each intermediate image.
display(image)
```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/ironman.jpg" alt="min-dalle" width="300"/>
```python ```python
text = 'court sketch of godzilla on trial' image_stream = model.generate_image_stream(
image = model.generate_image(text, grid_size=3) text='Dali painting of WALL·E',
display(image) seed=-1,
``` grid_size=3,
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/godzilla_trial.jpg" alt="min-dalle" width="300"/> log2_mid_count=3,
log2_supercondition_factor=3
)
```python is_first = True
text = 'a funeral at Whole Foods' for image in image_stream:
image = model.generate_image(text, grid_size=3) display_image = display if is_first else update_display
display(image) display_image(image, display_id=1)
is_first = False
``` ```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/funeral.jpg" alt="min-dalle" width="300"/> <img src="https://github.com/kuprel/min-dalle/raw/main/examples/dali_walle_animated.gif" alt="min-dalle" width="300"/>
```python
text = 'Jesus turning water into wine on Americas Got Talent'
image = model.generate_image(text, grid_size=3)
display(image)
```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/jesus.jpg" alt="min-dalle" width="300"/>
```python
text = 'cctv footage of Yoda robbing a liquor store'
image = model.generate_image(text, grid_size=3)
display(image)
```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/yoda.jpg" alt="min-dalle" width="300"/>
### Command Line ### Command Line
@ -85,8 +78,3 @@ Use `image_from_text.py` to generate images from the command line.
$ python image_from_text.py --text='artificial intelligence' --no-mega $ python image_from_text.py --text='artificial intelligence' --no-mega
``` ```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/artificial_intelligence.jpg" alt="min-dalle" width="200"/> <img src="https://github.com/kuprel/min-dalle/raw/main/examples/artificial_intelligence.jpg" alt="min-dalle" width="200"/>
```bash
$ python image_from_text.py --text='trail cam footage of gollum eating watermelon' --mega --grid-size=3
```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/gollum_trailcam.jpg" alt="min-dalle" width="300"/>

BIN
examples/dalle_walle_animated.gif vendored Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 1.5 MiB