Update README.md
Browse files
README.md
CHANGED
@@ -33,4 +33,20 @@ output = model(tokenizer('a b c d a b c').unsqueeze(0))
|
|
33 |
print(tokenizer.decode(output.argmax(2)))
|
34 |
|
35 |
>>> [['bos', 'a', 'b', 'c', 'd', 'a', 'b', 'c']]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
```
|
|
|
33 |
print(tokenizer.decode(output.argmax(2)))
|
34 |
|
35 |
>>> [['bos', 'a', 'b', 'c', 'd', 'a', 'b', 'c']]
|
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 |
```
|