Skip to content

Commit

Permalink
readme file added
Browse files Browse the repository at this point in the history
  • Loading branch information
meet-rocking committed Dec 20, 2024
1 parent bc8caad commit 0b4d282
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions frontend/core/pipelines/stable-diffusion-2/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
# Stable Diffusion Image Generation Pipeline

This repository contains a Python function `compute` to generate images based on textual descriptions using the Stable Diffusion model. The function utilizes the `diffusers` library and GPU acceleration for efficient image generation.

## Features

- Generate high-quality images from textual prompts.
- Supports inference using the Euler Discrete Scheduler for diffusion.
- Utilizes GPU acceleration with `float16` for enhanced performance.

---

## Usage
The compute function accepts a textual description (prompt) and the number of inference steps to generate an image. The generated image is saved as result.png in the current working directory.

## Function Parameters
- prompt (str): Text description of the desired image.
- inference_steps (int): Number of diffusion steps for the generation process.
- Return Value:
The function returns a dictionary containing the path to the generated image:

## Model used
- https://huggingface.co/stabilityai/stable-diffusion-2

0 comments on commit 0b4d282

Please sign in to comment.