Junior Studio

Trained on professional studio portraits of kids.

Details

• Type: LoRA

• Base Model: Stable Diffusion XL 1.0

• VAE: SDXL-VAE-FP16-Fix

• Trigger Word: junior studio

Sources

This model is available on:

• Hugging Face

• Civitai

Usage

To just use the base model, you can run:

import torch
from diffusers import DiffusionPipeline, AutoencoderKL
device = "cuda" if torch.cuda.is_available() else "cpu"
vae = AutoencoderKL.from_pretrained('madebyollin/sdxl-vae-fp16-fix', torch_dtype=torch.float16)
pipe = DiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    vae=vae, torch_dtype=torch.float16, variant="fp16",
    use_safetensors=True
)
pipe.to(device)
# This is where you load your trained weights
specific_safetensors = "pytorch_lora_weights.safetensors"
lora_scale = 0.9
pipe.load_lora_weights(
    'dwancin/junior-studio', 
    weight_name = specific_safetensors,
    # use_auth_token = True 
)
prompt = "A junior studio portrait"
image = pipe(
    prompt=prompt, 
    num_inference_steps=50,
    cross_attention_kwargs={"scale": lora_scale}
).images[0]

Dependencies

Make sure to upgrade diffusers to >= 0.19.0:

pip install diffusers --upgrade

In addition make sure to install transformers, safetensors, accelerate as well as the invisible watermark:

pip install invisible_watermark transformers accelerate safetensors