beyond commited on
Commit
9e98072
1 Parent(s): eb4b8d2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -52,4 +52,14 @@ sketch = "<mask> Conference on Empirical Methods <mask> submission of research p
52
  # 3. here we go!
53
  generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
54
  print(generated_text)
 
 
 
 
 
 
 
 
 
 
55
  ```
 
52
  # 3. here we go!
53
  generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
54
  print(generated_text)
55
+ ```
56
+
57
+ If you find our paper/code/demo useful, please cite our paper:
58
+ ```
59
+ @article{guo2022genius,
60
+ title={GENIUS: Sketch-based Language Model Pre-training via Extreme and Selective Masking for Text Generation and Augmentation},
61
+ author={Guo, Biyang and Gong, Yeyun and Shen, Yelong and Han, Songqiao and Huang, Hailiang and Duan, Nan and Chen, Weizhu},
62
+ journal={arXiv preprint arXiv:2211.10330},
63
+ year={2022}
64
+ }
65
  ```