Views
No views yet
1from diffusers import StableDiffusionPipeline
2import torch
3
4model_path = "Sourabh2/Human_Face_generator"
5pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
6pipe.to("cuda")
7
8prompt = "a man with blonde hair and beard"
9
10image = pipe(prompt=prompt).images[0]
11image.save("face.png")