Update README.md
Browse files
README.md
CHANGED
@@ -32,4 +32,21 @@ tokenizer = Tokenizer(model.config.input_encoding_map, model.config.output_encod
|
|
32 |
output = model(tokenizer('r e v e r s e').unsqueeze(0))
|
33 |
print(tokenizer.decode(output.argmax(2)))
|
34 |
>>> [['bos', 'e', 's', 'r', 'e', 'v', 'e', 'r']]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
```
|
|
|
32 |
output = model(tokenizer('r e v e r s e').unsqueeze(0))
|
33 |
print(tokenizer.decode(output.argmax(2)))
|
34 |
>>> [['bos', 'e', 's', 'r', 'e', 'v', 'e', 'r']]
|
35 |
+
```
|
36 |
+
|
37 |
+
|
38 |
+
|
39 |
+
### 🙏Cite🙏
|
40 |
+
|
41 |
+
|
42 |
+
###### If you are interested in our paper, please feel free to cite it.
|
43 |
+
```
|
44 |
+
@misc{weng2023neural,
|
45 |
+
title={Neural Comprehension: Language Models with Compiled Neural Networks},
|
46 |
+
author={Yixuan Weng and Minjun Zhu and Fei Xia and Bin Li and Shizhu He and Kang Liu and Jun Zhao},
|
47 |
+
year={2023},
|
48 |
+
eprint={2304.01665},
|
49 |
+
archivePrefix={arXiv},
|
50 |
+
primaryClass={cs.CL}
|
51 |
+
}
|
52 |
```
|