sdxl-panorama LoRA by jbilcke
First version of my panorama LoRA (use 1024x512)
Inference with Replicate API
Grab your replicate token here
pip install replicate
export REPLICATE_API_TOKEN=r8_*************************************
import replicate
output = replicate.run(
"sdxl-panorama@sha256:fd1f53277a424000168dfceeef58934c43bb3c27a3843a75e7b6d72c46cfc8fa",
input={"prompt": "hdri view, a nice house in London, in the style of TOK"}
)
print(output)
You may also do inference via the API with Node.js or curl, and locally with COG and Docker, check out the Replicate API page for this model
Inference with 𧨠diffusers
Replicate SDXL LoRAs are trained with Pivotal Tuning, which combines training a concept via Dreambooth LoRA with training a new token with Textual Inversion.
As diffusers
doesn't yet support textual inversion for SDXL, we will use cog-sdxl TokenEmbeddingsHandler
class.
The trigger tokens for your prompt will be <s0><s1>
pip install diffusers transformers accelerate safetensors huggingface_hub
git clone https://github.com/replicate/cog-sdxl cog_sdxl
import torch
from huggingface_hub import hf_hub_download
from diffusers import DiffusionPipeline
from cog_sdxl.dataset_and_utils import TokenEmbeddingsHandler
from diffusers.models import AutoencoderKL
pipe = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0",
torch_dtype=torch.float16,
variant="fp16",
).to("cuda")
pipe.load_lora_weights("jbilcke-hf/sdxl-panorama", weight_name="lora.safetensors")
text_encoders = [pipe.text_encoder, pipe.text_encoder_2]
tokenizers = [pipe.tokenizer, pipe.tokenizer_2]
embedding_path = hf_hub_download(repo_id="jbilcke-hf/sdxl-panorama", filename="embeddings.pti", repo_type="model")
embhandler = TokenEmbeddingsHandler(text_encoders, tokenizers)
embhandler.load_embeddings(embedding_path)
prompt="hdri view, a nice house in London, in the style of <s0><s1>"
images = pipe(
prompt,
cross_attention_kwargs={"scale": 0.8},
).images
#your output image
images[0]
- Downloads last month
- 252
Model tree for jbilcke-hf/sdxl-panorama
Base model
stabilityai/stable-diffusion-xl-base-1.0