|
--- |
|
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. |
