Update README.md

README.md

@@ -1,10 +1,25 @@
+---
+language:
+- en
+tags:
+- reco
+- text-to-image
+- layout-to-image
+pipeline_tag: text-to-image
+---
 
-# Diffusers 🧨 port of [ReCo: Region-Controlled Text-to-Image Generation (CVPR 2023)](https://github.com/microsoft/ReCo)
+
+# Diffusers 🧨 port of [ReCo: Region-Controlled Text-to-Image Generation (CVPR 2023)](https://arxiv.org/abs/2211.15518)
 
 - Original authors: Zhengyuan Yang, Jianfeng Wang, Zhe Gan, Linjie Li, Kevin Lin, Chenfei Wu, Nan Duan, Zicheng Liu, Ce Liu, Michael Zeng, Lijuan Wang
 - Original github repo by authors: https://github.com/microsoft/ReCo
 - Converted to Diffusers: Jaemin Cho
 
+
+# COCO checkpoint
+- original pytorch lightning checkpoint: https://unitab.blob.core.windows.net/data/reco/reco_coco_616.ckpt
+- original configuration yaml: https://github.com/microsoft/ReCo/blob/main/configs/reco/v1-finetune_cocogit.yaml
+
 # Example Usage
 
 ```python
@@ -92,5 +107,4 @@ boxes = [[263.68, 294.912, 380.544, 392.832], [121.344, 265.216, 267.392, 401.92
 prompt = create_reco_prompt(caption, phrases, boxes)
 prompt
 >>> 'A box contains six donuts with varying types of glazes and toppings. <|endoftext|> <bin514> <bin575> <bin743> <bin766> <|startoftext|> chocolate donut. <|endoftext|> <bin237> <bin517> <bin522> <bin784> <|startoftext|> dark vanilla donut. <|endoftext|> <bin763> <bin575> <bin988> <bin745> <|startoftext|> donut with sprinkles. <|endoftext|> <bin234> <bin281> <bin524> <bin527> <|startoftext|> donut with powdered sugar. <|endoftext|> <bin515> <bin259> <bin767> <bin514> <|startoftext|> pink donut. <|endoftext|> <bin753> <bin290> <bin957> <bin506> <|startoftext|> brown donut. <|endoftext|>'
-```
-
+```