diffusion_fashion / README.md
MohamedRashad's picture
Update README.md
aa429da
|
raw
history blame
1.05 kB
metadata
language: en
tags:
  - stable-diffusion
  - diffusers
  - text-to-image
  - fashion
  - diffusion
  - openjourney
inference: true
license: openrail
library_name: diffusers

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