weights

README.md

@@ -1,3 +1,37 @@
 ---
 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

checkpoint-1200/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c70512cf37d4601bcf13d5df5f59a5c6c9bd680877b4b4feea27f5a29a5f4dc
+size 3417037

checkpoint-15000/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba1ed76ebe9dcd17839b915c1b5c964533695fa157d44eb8c1f0e8617347edaa
+size 3417037

checkpoint-18000/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:305c0d5bbff288945b75b04d75ae05135450ef9a7ede50f7d373535ce1a31892
+size 3417037

checkpoint-6000/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b317cf7f338b7c40505c4af50f229e657708e708234762a517c151769f1e428e
+size 3417037

checkpoint-9000/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:805c4813c12768d8e6427532ff82f2740a611afc9a6f7d407acd61ba18ec9e1d
+size 3417037

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba1ed76ebe9dcd17839b915c1b5c964533695fa157d44eb8c1f0e8617347edaa
+size 3417037