Views
No views yet
1from diffusers import StableDiffusionInpaintPipeline
2import torch
3
4pipe = StableDiffusionInpaintPipeline.from_pretrained(
5 "stabilityai/stable-diffusion-2-inpainting",
6 torch_dtype=torch.float16,
7 device_map="auto",
8 cache_dir=CACHE_DIR,
9)
10pipe.unet.load_attn_procs(
11 lora_model,
12 use_safetensors=False,
13)