Views
No views yet
1from diffusers import StableDiffusionPipeline
2import torch
3
4pipeline = StableDiffusionPipeline.from_pretrained("MohamedRashad/diffusion_fashion", torch_dtype=torch.float16)
5pipeline.to("cuda")
6
7prompt = "A photo of a dress, made in 2019, color is Red, Casual usage, Women's cloth, something for the summer season, on white background"
8images = pipeline(prompt).images[0]
9image.save("red_dress.png")
10