File size: 829 Bytes
834f33b e4bc8da 834f33b e4bc8da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
---
license: cc-by-nc-4.0
base_model: stabilityai/stable-diffusion-2-inpainting
dataset: sshh12/sentinel-2-rgb-captioned
tags:
- stable-diffusion
- text-to-image
- diffusers
- lora
- inpainting
- procedural-generation
inference: false
---
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
```py
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
GitHub: https://github.com/sshh12/terrain-diffusion
|