TCD SD15 LoRA

🚀 TCD LoRA for Stable Diffusion v1.5

論文「Trajectory Consistency Distillation」のStable Diffusion v1.5用の公式TCD LoRAです。画像生成において高品質な結果を提供します。

🚀 クイックスタート

論文「Trajectory Consistency Distillation」(https://arxiv.org/abs/2402.19159) のStable Diffusion v1.5用の公式TCD LoRAです。

詳細な使い方については、プロジェクトページ を参照してください。

💻 使用例

基本的な使用法

import torch
from diffusers import  StableDiffusionPipeline, TCDScheduler
device = "cuda"
base_model_id = "runwayml/stable-diffusion-v1-5"
tcd_lora_id = "h1t/TCD-SD15-LoRA"
pipe = StableDiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16, variant="fp16").to(device)
pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
pipe.load_lora_weights(tcd_lora_id)
pipe.fuse_lora()
prompt = "Beautiful woman, bubblegum pink, lemon yellow, minty blue, futuristic, high-detail, epic composition, watercolor."
image = pipe(
    prompt=prompt,
    num_inference_steps=4,
    guidance_scale=0,
    # Eta (referred to as `gamma` in the paper) is used to control the stochasticity in every step.
    # A value of 0.3 often yields good results.
    # We recommend using a higher eta when increasing the number of inference steps.
    eta=0.3, 
    generator=torch.Generator(device=device).manual_seed(42),
).images[0]

📄 ライセンス

このモデルはMITライセンスの下で提供されています。