ReginaldIII t1_ir067hc wrote on October 4, 2022 at 11:46 AM

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

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)

DuLLSoN t1_ir0esd1 wrote on October 4, 2022 at 1:03 PM

This took 23 seconds on my T4 GPU Colab test.

Not a bad time, but nowhere near 13 seconds.

ReginaldIII t1_ir0i174 wrote on October 4, 2022 at 1:29 PM

Yeah, fair enough I just tried it again. Maybe they randomly gave me a more beefy GPU last night. Still 13 seconds for 3.