Viewing a single comment thread. View all comments

DuLLSoN t1_ir04t67 wrote

Reply to comment by ReginaldIII in [N] Stable Diffusion reaches new record (with explanation + colab link) by Norlax_42

5x 512x512 images with 25 steps on T4 Colab in 13 seconds? I would like to see a notebook of that.

I wonder if you mean 13 seconds per image because this implementation reports ~10s per image with mixed precision.

2

ReginaldIII t1_ir067hc wrote 

import keras_cv
from tensorflow import keras

keras.mixed_precision.set_global_policy("mixed_float16")
model = keras_cv.models.StableDiffusion(img_width=512, img_height=512, jit_compile=True)

images = model.text_to_image("photograph of an astronaut riding a horse", batch_size=5)
3