-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
bc8caad
commit 0b4d282
Showing
1 changed file
with
23 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |