Skip to content

Commit be78e00

Browse files
authored
Render StableDiffusion guide on an A100 GPU (keras-team#1087)
* Re-render with fixed prompt * fix lack of warmup * Updated rendering
1 parent a5da87a commit be78e00

11 files changed

+1152
-3
lines changed
1.26 MB
Loading
949 KB
Loading
1.06 MB
Loading
1.36 MB
Loading
1.08 MB
Loading
980 KB
Loading
1.05 MB
Loading
933 KB
Loading

guides/ipynb/keras_cv/generate_images_with_stable_diffusion.ipynb

Lines changed: 640 additions & 0 deletions
Large diffs are not rendered by default.

guides/keras_cv/generate_images_with_stable_diffusion.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -284,9 +284,7 @@ def plot_images(images):
284284
model = keras_cv.models.StableDiffusion(jit_compile=True)
285285
# Before we benchmark the model, we run inference once to make sure the TensorFlow
286286
# graph has already been traced.
287-
images = model.text_to_image(
288-
"An oldschool macintosh computer showing an avocado on its screen",
289-
)
287+
images = model.text_to_image("An avocado armchair", batch_size=3)
290288
plot_images(images)
291289

292290
"""

0 commit comments

Comments
 (0)