İnpainting LoRA modeli

Bu model, görüntülerdeki eksik kısımları (maskelenmiş bölgeleri) doldurmak için eğitilmiş bir LoRA modelidir.

Model Açıklaması

Taban model: stabilityai/stable-diffusion-2-base
LoRA type: UNet attention layers (to_q, to_k, to_v, to_out.0)
İnpainting: Görüntülerdeki maskelenmiş bölgeleri akıllıca tamamlar

Kullanım

Bu model, görüntülerde istenmeyen öğeleri kaldırmak, eksik kısımları doldurmak veya var olan içeriği değiştirmek için kullanılabilir.

import torch
from diffusers import StableDiffusionImg2ImgPipeline
from peft import PeftModel

# Model yükleme
model_id = "stabilityai/stable-diffusion-2-base"
lora_model_id = "meryemarpaci/inpainting-lora" # Bu repo

pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
    model_id,
    torch_dtype=torch.float16
)
pipe.unet = PeftModel.from_pretrained(pipe.unet, lora_model_id)
pipe = pipe.to("cuda")

# Görüntü ve maske hazırlama
# ...

# İnpainting
result = pipe(
    prompt="Yüksek kaliteli bir fotoğraf",
    image=masked_image,
    strength=0.8,
    num_inference_steps=50,
    guidance_scale=7.5,
).images[0]

Web Arayüzü

Bu modeli kullanmak için aşağıdaki Hugging Face Space'i ziyaret edebilirsiniz: İnpainting Web UI

Örnekler

(İnpainting sonuç örnekleri buraya eklenebilir)

Downloads last month: 5