lego-sdxl-dora / README.md
merve's picture
merve HF staff
Update README.md
b452e15 verified
metadata
tags:
  - stable-diffusion-xl
  - stable-diffusion-xl-diffusers
  - text-to-image
  - diffusers
  - lora
  - template:sd-lora
widget:
  - text: a lego set in the style of <s0><s1>, an astronaut riding a horse
    output:
      url: image_0.png
  - text: a lego set in the style of <s0><s1>, an astronaut riding a horse
    output:
      url: image_1.png
  - text: a lego set in the style of <s0><s1>, an astronaut riding a horse
    output:
      url: image_2.png
  - text: a lego set in the style of <s0><s1>, an astronaut riding a horse
    output:
      url: image_3.png
base_model: stabilityai/stable-diffusion-xl-base-1.0
instance_prompt: lego set in the style of <s0><s1>
license: openrail++

SDXL LoRA DreamBooth - merve/lego-sdxl-dora

Prompt
a lego set in the style of <s0><s1>, an astronaut riding a horse
Prompt
a lego set in the style of <s0><s1>, an astronaut riding a horse
Prompt
a lego set in the style of <s0><s1>, an astronaut riding a horse
Prompt
a lego set in the style of <s0><s1>, an astronaut riding a horse

Model description

These are merve/lego-sdxl-dora LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.

Download model

Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke

  • LoRA: download lego-sdxl-dora.safetensors here 💾.
    • Place it on your models/Lora folder.
    • On AUTOMATIC1111, load the LoRA by adding <lora:lego-sdxl-dora:1> to your prompt. On ComfyUI just load it as a regular LoRA.
  • Embeddings: download lego-sdxl-dora_emb.safetensors here 💾.
    • Place it on it on your embeddings folder
    • Use it by adding lego-sdxl-dora_emb to your prompt. For example, lego set in the style of lego-sdxl-dora_emb (you need both the LoRA and the embeddings as they were trained together for this LoRA)

Use it with the 🧨 diffusers library

from diffusers import AutoPipelineForText2Image
import torch
from huggingface_hub import hf_hub_download
from safetensors.torch import load_file
        
pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
pipeline.load_lora_weights('merve/lego-sdxl-dora', weight_name='pytorch_lora_weights.safetensors')
embedding_path = hf_hub_download(repo_id='merve/lego-sdxl-dora', filename='lego-sdxl-dora_emb.safetensors', repo_type="model")
state_dict = load_file(embedding_path)
pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
        
image = pipeline('a lego set in the style of <s0><s1>, an astronaut riding a horse').images[0]

For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers

Trigger words

To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:

to trigger concept TOK → use <s0><s1> in your prompt

Details

All Files & versions.

The weights were trained using 🧨 diffusers Advanced Dreambooth Training Script.

LoRA for the text encoder was enabled. False.

Pivotal tuning was enabled: True.

Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.