Update README.md

README.md

@@ -1,3 +1,26 @@
----
-license: mit
----
+---
+license: mit
+datasets:
+- unography/SAM-LLAVA-55k
+language:
+- en
+library_name: diffusers
+pipeline_tag: image-to-image
+tags:
+- stable-diffusion
+---
+# Model Card for lavinal712/controlnet-segmentation-sam-llava
+
+## Model Description
+
+This is controlnet weight trained on runwayml/stable-diffusion-v1-5 with segmentaion.
+
+### Training
+
+This model was trained using a Segmented dataset based on the SAM-LLaVA-Captions10M Dataset. Stable Diffusion v1.5 checkpoint was used as the base model for the controlnet.
+
+- [SAM-LLAVA-55k](https://huggingface.co/datasets/unography/SAM-LLAVA-55k)
+
+#### Training Method
+
+- Train on [SAM-LLAVA-55k](https://huggingface.co/datasets/unography/SAM-LLAVA-55k) for 60000 steps with batch size of 4.