From cc9bdb8f7b19b02903ef0d56d341fa0f6e555010 Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Fri, 1 Jul 2022 19:08:42 -0400 Subject: [PATCH] update readme --- README.md | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index cccb956..53edc1e 100644 --- a/README.md +++ b/README.md @@ -5,9 +5,9 @@ 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 takes **7.4 seconds** to generate an image with DALL·E Mega on a standard GPU runtime in Colab. -The flax model, and the code for coverting it to torch can be found [here](https://github.com/kuprel/min-dalle-flax). +The flax model and the code for coverting it to torch can be found [here](https://github.com/kuprel/min-dalle-flax). ## Install @@ -32,9 +32,10 @@ $ python image_from_text.py --text='court sketch of godzilla on trial' --mega ``` ![Godzilla Trial](examples/godzilla_on_trial.png) + ### Python -To load a model once and generate multiple times, first initialize `MinDalleTorch` +To load a model once and generate multiple times, first initialize `MinDalleTorch`. ```python from min_dalle import MinDalleTorch @@ -46,19 +47,19 @@ model = MinDalleTorch( ) ``` + The required models will be downloaded to `models_root` if they are not already there. After the model has loaded, call `generate_image` with some text and a seed as many times as you want. ```python -image = model.generate_image("a comfy chair that looks like an avocado") +text = "a comfy chair that looks like an avocado" +image = model.generate_image(text) display(image) ``` ![Avocado Armchair](examples/avocado_armchair.png) ```python -image = model.generate_image( - "trail cam footage of gollum eating watermelon", - seed=1 -) +text = "trail cam footage of gollum eating watermelon" +image = model.generate_image(text, seed=1) display(image) ``` ![Gollum Trailcam](examples/gollum_trailcam.png) \ No newline at end of file