diff --git a/README.md b/README.md index 922eada..7e9b678 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ 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. -It currently takes **7.4 seconds** to generate an image with DALL·E Mega on a standard GPU runtime in Colab. +It currently take **35 seconds** to generate a 3x3 grid with DALL·E Mega on a standard GPU runtime in Colab. The flax model and code for converting it to torch can be found [here](https://github.com/kuprel/min-dalle-flax). @@ -41,7 +41,7 @@ display(image) ```python text = "trail cam footage of gollum eating watermelon" -image = model.generate_image(text, seed=1) +image = model.generate_image(text, seed=1, grid_size=3) display(image) ``` ![Gollum Trailcam](https://github.com/kuprel/min-dalle/raw/main/examples/gollum_trailcam.png) diff --git a/examples/gollum_trailcam.png b/examples/gollum_trailcam.png index fdb3d0b..56be0b7 100644 Binary files a/examples/gollum_trailcam.png and b/examples/gollum_trailcam.png differ