diff --git a/README.md b/README.md index 8341c2c..615d6f4 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 minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini) in PyTorch. It has been stripped to the bare essentials necessary for doing inference. The only third party dependencies are numpy and torch. +This is a minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini) in PyTorch. It has been stripped to the bare essentials necessary for doing inference. The only third party dependencies are numpy, requests, pillow and torch. It currently take **35 seconds** to generate a 3x3 grid with DALL·E Mega on a standard GPU runtime in Colab. @@ -59,4 +59,4 @@ $ python image_from_text.py --text='artificial intelligence' --seed=7 ```bash $ python image_from_text.py --text='trail cam footage of gollum eating watermelon' --mega --seed=1 --grid-size=3 ``` -![Gollum Trailcam](https://github.com/kuprel/min-dalle/raw/main/examples/gollum_trailcam.png) \ No newline at end of file +![Gollum Trailcam](https://github.com/kuprel/min-dalle/raw/main/examples/gollum_trailcam.png) diff --git a/requirements.txt b/requirements.txt index 326574e..4d2dd2b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,4 @@ min-dalle +numpy==1.23.0 +pillow==9.2.0 +requests==2.28.1