Inference
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/sdxl-lora-finetune-product-caption")
prompt = "outer, The Nike x Balenciaga Down Jacket Black, a 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")
- Downloads last month
- 94
Model tree for VikramSingh178/sdxl-lora-finetune-product-caption
Base model
stabilityai/stable-diffusion-xl-base-1.0