Views
No views yet
1import torch
2from diffusers import StableDiffusionImg2ImgPipeline
3from peft import PeftModel
4
5# Model yükleme
6model_id = "stabilityai/stable-diffusion-2-base"
7lora_model_id = "meryemarpaci/sd2base-inpainting-lora" # Bu repo
8
9pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
10 model_id,
11 torch_dtype=torch.float16
12)
13pipe.unet = PeftModel.from_pretrained(pipe.unet, lora_model_id)
14pipe = pipe.to("cuda")
15
16# Görüntü ve maske hazırlama
17# ...
18
19# İnpainting
20result = pipe(
21 prompt="Yüksek kaliteli bir fotoğraf",
22 image=masked_image,
23 strength=0.8,
24 num_inference_steps=50,
25 guidance_scale=7.5,
26).images[0]