Update README.md
Browse files
README.md
CHANGED
|
@@ -33,4 +33,20 @@ output = model(tokenizer('w e n g s y x').unsqueeze(0))
|
|
| 33 |
print(tokenizer.decode(output.argmax(2)))
|
| 34 |
|
| 35 |
>>> [['bos', 'x', 'x', 'x', 'x', 'x', 'x', 'x']]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
```
|
|
|
|
| 33 |
print(tokenizer.decode(output.argmax(2)))
|
| 34 |
|
| 35 |
>>> [['bos', 'x', 'x', 'x', 'x', 'x', 'x', 'x']]
|
| 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 |
```
|