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-03 15:40:58 -04:00
parent e20c4a8c9f
commit 948e05c0dd
3 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
README.md vendored
View File

@ -6,7 +6,7 @@
   
[![Join us on Discord](https://img.shields.io/discord/823813159592001537?color=5865F2&logo=discord&logoColor=white)](https://discord.gg/xBPBXfcFHd) [![Join us on Discord](https://img.shields.io/discord/823813159592001537?color=5865F2&logo=discord&logoColor=white)](https://discord.gg/xBPBXfcFHd)
This is a fast, minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini). It has been stripped for inference and converted to PyTorch. The only third party dependencies are numpy, requests, pillow and torch. This is a fast, minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini). It has been stripped down for inference and converted to PyTorch. The only third party dependencies are numpy, requests, pillow and torch.
To generate a 3x3 grid of DALL·E Mega images it takes To generate a 3x3 grid of DALL·E Mega images it takes
- **35 seconds** with a P100 in Colab - **35 seconds** with a P100 in Colab
@ -23,8 +23,6 @@ $ pip install min-dalle
## Usage ## Usage
### Python
Load the model parameters once and reuse the model to generate multiple images. Load the model parameters once and reuse the model to generate multiple images.
```python ```python
@ -43,11 +41,11 @@ display(image)
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/avocado_armchair.png" alt="drawing" width="200"/> <img src="https://github.com/kuprel/min-dalle/raw/main/examples/avocado_armchair.png" alt="drawing" width="200"/>
```python ```python
text = 'court sketch of godzilla on trial' text = 'Dali painting of WallE'
image = model.generate_image(text, seed=6, grid_size=3) image = model.generate_image(text, seed=0, grid_size=3)
display(image) display(image)
``` ```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/godzilla_trial.png" alt="drawing" width="300"/> <img src="https://github.com/kuprel/min-dalle/raw/main/examples/dali_walle.png" alt="drawing" width="300"/>
```python ```python
text = 'Rusty Iron Man suit found abandoned in the woods being reclaimed by nature' text = 'Rusty Iron Man suit found abandoned in the woods being reclaimed by nature'
@ -56,6 +54,13 @@ display(image)
``` ```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/ironman.png" alt="drawing" width="300"/> <img src="https://github.com/kuprel/min-dalle/raw/main/examples/ironman.png" alt="drawing" width="300"/>
```python
text = 'court sketch of godzilla on trial'
image = model.generate_image(text, seed=6, grid_size=3)
display(image)
```
<img src="https://github.com/kuprel/min-dalle/raw/main/examples/godzilla_trial.png" alt="drawing" width="300"/>
```python ```python
text = 'a funeral at Whole Foods' text = 'a funeral at Whole Foods'
image = model.generate_image(text, seed=10, grid_size=3) image = model.generate_image(text, seed=10, grid_size=3)

BIN
examples/dali_walle.png vendored Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 1.1 MiB

4
setup.py
View File

@ -16,7 +16,9 @@ setuptools.setup(
license='MIT', license='MIT',
install_requires=[ install_requires=[
'torch>=1.10.0', 'torch>=1.10.0',
'typing_extensions>=4.1.0' 'typing_extensions>=4.1.0',
'numpy>=1.21',
'pillow>=7.1'
], ],
keywords = [ keywords = [
'artificial intelligence', 'artificial intelligence',