Hugging Face's logo

bharadwajkg
/
finetune-sd2-1-planogram-lora-data5

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
lora
Model card Files
xet
 Metrics Community
How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("bharadwajkg/finetune-sd2-1-planogram-lora-data5")

prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]

LoRA text2image fine-tuning - bharadwajkg/finetune-sd2-1-planogram-lora-data5

These are LoRA adaption weights for stabilityai/stable-diffusion-2-1. The weights were fine-tuned on the bharadwajkg/planogram-sd-data5 dataset. You can find some example images in the following.

img_0 img_1 img_2 img_3

Downloads last month
1
Inference Providers NEW
Text-to-Image
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for bharadwajkg/finetune-sd2-1-planogram-lora-data5

Base model

stabilityai/stable-diffusion-2-1
Adapter
(340)
this model