Modelpublic

runchat/lora-533d7b31-63fd-42a0-be75-b68de7db171f-shell

sourceHugging Faceopenrail++updated 1y agoView on Hugging Face

0likes3downloads

Model Card

SDXL LoRA: shell

This is a LoRA (Low-Rank Adaptation) model for Stable Diffusion XL fine-tuned on images with the trigger word shell.

Usage

python

from diffusers import StableDiffusionXLPipeline
import torch

# Load base model
pipe = StableDiffusionXLPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    torch_dtype=torch.float16
)

# Load LoRA weights
pipe.load_lora_weights("runchat/lora-533d7b31-63fd-42a0-be75-b68de7db171f-shell")
pipe = pipe.to("cuda")

# Generate image
prompt = "a photo of shell style"
image = pipe(prompt, num_inference_steps=25, guidance_scale=7.5).images[0]
image.save("output.png")

Training Details

—Base model: stabilityai/stable-diffusion-xl-base-1.0
—Training steps: 1000
—Learning rate: 0.0001
—Batch size: 1
—LoRA rank: 32
—Trigger word: shell