VikramSingh178's picture
Update README.md
c6a3c5b verified
metadata
license: creativeml-openrail-m
library_name: diffusers
tags:
  - stable-diffusion-xl
  - stable-diffusion-xl-diffusers
  - text-to-image
  - diffusers
  - diffusers-training
base_model: stabilityai/stable-diffusion-xl-base-1.0
inference: true
datasets:
  - VikramSingh178/Products-10k-BLIP-captions
language:
  - en

SDXL Product Images

How to use

from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
pipe.to("cuda")
pipe.load_lora_weights("VikramSingh178/Products10k-SDXL-Lora)

prompt = "Commercial photography of a black down jacket with a logo on the chest."

image = pipe(prompt, num_inference_steps=45, guidance_scale=7.5).images[0]
image.save("example.png")