Image Generation

9 minutes

Running Stable Diffusion Locally

1. Introduction

Wanted to test the RTX 5050 (8 GB) GPU on my laptop. So thought of running a Stable Diffusion Model for image generation. Since I was exploring TensorFlow for sometime, picked up this guide High-performance image generation using Stable Diffusion in KerasCV for a step by step guidance.

2. Stable Diffusion

We will use the KerasCV implementation of the Stable Diffusion Model. Stable Diffusion is an open-source text-to-image model which generate images based on a text prompt.