Update README.md
Browse files
README.md
CHANGED
|
@@ -7,4 +7,6 @@ pipeline_tag: text-to-image
|
|
| 7 |
tags:
|
| 8 |
- stable-diffusion
|
| 9 |
---
|
| 10 |
-
|
|
|
|
|
|
|
|
|
| 7 |
tags:
|
| 8 |
- stable-diffusion
|
| 9 |
---
|
| 10 |
+
# TokenCompose SD14 Model Card
|
| 11 |
+
|
| 12 |
+
[TokenCompose_SD14_A](https://mlpc-ucsd.github.io/TokenCompose/) is a [latent text-to-image diffusion model](https://arxiv.org/abs/2112.10752) finetuned from the [**Stable-Diffusion-v1-4**](https://huggingface.co/CompVis/stable-diffusion-v1-4) checkpoint at resolution 512x512 on the [VSR](https://github.com/cambridgeltl/visual-spatial-reasoning) split of [COCO image-caption pairs](https://cocodataset.org/#download) for 24,000 steps with a learning rate of 5e-6. The training objective involves token-level grounding terms in addition to denoising loss for enhanced multi-category instance composition and photorealism.
|