Sourabh2's picture
Update README.md
a4e583e verified
|
raw
history blame
465 Bytes

language: en

Example Usage

from diffusers import StableDiffusionPipeline
import torch

model_path = "Sourabh2/Human_Face_generator"
pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
pipe.to("cuda")

prompt = "a man with blonde hair and beard"

image = pipe(prompt=prompt).images[0]
image.save("face.png")

Example Output

pareto