Update README.md

README.md

@@ -34,7 +34,7 @@ inference:
 
 ![sega-illustration](https://cdn.jsdelivr.net/gh/beyondguo/mdnice_pictures/typora/sega-main-illustration.png)
 
-- Paper: [coming soon](to_be_added)
+- Paper: [SEGA: SkEtch-based Generative Augmentation (preprint)](https://github.com/beyondguo/SEGA/blob/master/SEGA_gby_preprint.pdf)
 - GitHub: [SEGA](https://github.com/beyondguo/SEGA). 
 
 **SEGA中文版** 可以根据你给出的一个**草稿**进行填词造句扩写，草稿可以是：
@@ -68,12 +68,13 @@ for sketch in sketchs:
 
 ## Model variations / SEGA其他版本
 
-| Model | #params | Language |
-|------------------------|--------------------------------|-------|
-| [`sega-large`](https://huggingface.co/beyond/sega-large) | xM   | English |
-| [`sega-base`(coming soon)]()  | xM    | English |
-| [`sega-large-chinese`(coming soon)]() | xM    |  Chinese |
-| [`sega-base-chinese`](https://huggingface.co/beyond/sega-base-chinese) | xM    | Chinese |
+| Model | #params | Language | comment|
+|------------------------|--------------------------------|-------|---------|
+| [`sega-large`](https://huggingface.co/beyond/sega-large) | 406M   | English | The version used in paper |
+| [`sega-large-k2t`](https://huggingface.co/beyond/sega-large-k2t)  | 406M    | English | keywords-to-text |
+| [`sega-base`](https://huggingface.co/beyond/sega-base)  | 139M    | English | smaller version |
+| [`sega-base-ps`](https://huggingface.co/beyond/sega-base)  | 139M    | English | pre-trained both in paragraphs and short sentences |
+| [`sega-base-chinese`](https://huggingface.co/beyond/sega-base-chinese) | 116M    | 中文 | 在一千万纯净中文段落上预训练|
 
 
 ## Comparison / 效果对比