samikhao/my-stars-lora-sdxl

MY-STARS LoRA (SDXL)

LoRA-адаптер для стиля MY-STARS. Тренирован train_dreambooth_lora_sdxl.py (diffusers).

Загрузка в Diffusers

import torch
from diffusers import StableDiffusionXLPipeline

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

pipe.load_lora_weights("samikhao/my-stars-lora-sdxl", weight_name="pytorch_lora_weights.safetensors")

prompt = "a star in MY-STARS style"
image  = pipe(prompt, num_inference_steps=30, guidance_scale=5.0).images[0]
image.save("sample.png")
Основной файл: pytorch_lora_weights.safetensors (из checkpoint-500).