diff --git a/README.md b/README.md index 1735b09..d5080f8 100644 --- a/README.md +++ b/README.md @@ -6,7 +6,7 @@   [![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 - **35 seconds** with a P100 in Colab @@ -23,8 +23,6 @@ $ pip install min-dalle ## Usage -### Python - Load the model parameters once and reuse the model to generate multiple images. ```python @@ -43,11 +41,11 @@ display(image) drawing ```python -text = 'court sketch of godzilla on trial' -image = model.generate_image(text, seed=6, grid_size=3) +text = 'Dali painting of WallE' +image = model.generate_image(text, seed=0, grid_size=3) display(image) ``` -drawing +drawing ```python text = 'Rusty Iron Man suit found abandoned in the woods being reclaimed by nature' @@ -56,6 +54,13 @@ display(image) ``` drawing +```python +text = 'court sketch of godzilla on trial' +image = model.generate_image(text, seed=6, grid_size=3) +display(image) +``` +drawing + ```python text = 'a funeral at Whole Foods' image = model.generate_image(text, seed=10, grid_size=3) diff --git a/examples/dali_walle.png b/examples/dali_walle.png new file mode 100644 index 0000000..4ba1fe0 Binary files /dev/null and b/examples/dali_walle.png differ diff --git a/setup.py b/setup.py index ec38227..8625142 100644 --- a/setup.py +++ b/setup.py @@ -16,7 +16,9 @@ setuptools.setup( license='MIT', install_requires=[ 'torch>=1.10.0', - 'typing_extensions>=4.1.0' + 'typing_extensions>=4.1.0', + 'numpy>=1.21', + 'pillow>=7.1' ], keywords = [ 'artificial intelligence',