These are LoRA adaption weights for stabilityai/stable-diffusion-2-inpainting. The weights were fine-tuned on the sshh12/sentinel-2-rgb-captioned dataset.
🧨 Diffusers Usage
from diffusers import StableDiffusionInpaintPipeline
import torch
pipe = StableDiffusionInpaintPipeline.from_pretrained(
"stabilityai/stable-diffusion-2-inpainting",
torch_dtype=torch.float16,
device_map="auto",
cache_dir=CACHE_DIR,
)
pipe.unet.load_attn_procs(
lora_model,
use_safetensors=False,
)
Training
- Downloads last month
- 19
Model tree for sshh12/sd2-lora-inpainting-sentinel-2-rgb
Base model
stabilityai/stable-diffusion-2-inpainting