diffusion_fashion / README.md
MohamedRashad's picture
Update README.md
b1c4379
|
raw
history blame
904 Bytes
metadata
license: openrail

Stable Diffusion fine-tuned for Fashion Product Images Dataset

This model is a fine-tuned version of openjourney that is based on Stable Diffusion targeting fashion and clothing.

How to use ?

from diffusers import StableDiffusionPipeline
import torch

pipeline = StableDiffusionPipeline.from_pretrained("MohamedRashad/diffusion_fashion", torch_dtype=torch.float16)
pipeline.to("cuda")

prompt = "A photo of a dress, made in 2019, color is Red, Casual usage, Women's cloth, something for the summer season, on white background"
images = pipeline(prompt).images[0]
image.save("red_dress.png")

Any feedback or questions are welcomed on the community tab