Norm commited on
Commit
8d57af7
1 Parent(s): b8cee1d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -4,7 +4,14 @@ license: mit
4
 
5
  # ERNIE-Layout_Pytorch
6
 
7
- [This repository](https://github.com/NormXU/ERNIE-Layout-Pytorch) contains an unofficial ERNIE-Layout implementations in Pytorch, originally released via [PaddleNLP](https://github.com/PaddlePaddle/PaddleNLP). The model weight is converted from [PaddlePaddle/ernie-layoutx-base-uncased](https://huggingface.co/PaddlePaddle/ernie-layoutx-base-uncased) to PyTorch style with the [tools/convert2torch.py](https://github.com/NormXU/ERNIE-Layout-Pytorch/blob/main/tools/convert2torch.py) script. Feel free to edit it if necessary.
 
 
 
 
 
 
 
8
 
9
  **A Quick Example**
10
  ```python
@@ -56,5 +63,4 @@ answer = tokenizer.decode(encoding.input_ids[0][start_max: end_max])
56
  print(answer)
57
 
58
 
59
-
60
  ```
 
4
 
5
  # ERNIE-Layout_Pytorch
6
 
7
+ - **Model type:** [ERNIE-Layout](https://arxiv.org/abs/2210.06155)
8
+ - **Repository:** [source code](https://github.com/NormXU/ERNIE-Layout-Pytorch): an unofficial ERNIE-Layout implementation in Pytorch
9
+
10
+ - **Converted from:** [PaddlePaddle/ernie-layoutx-base-uncased](https://huggingface.co/PaddlePaddle/ernie-layoutx-base-uncased)
11
+
12
+
13
+ The ERNIE-Layout-Pytorch model is initially released by [PaddleNLP](https://github.com/PaddlePaddle/PaddleNLP). To make Pytorch users easy to use, the model has been converted into PyTorch format with the [tools/convert2torch.py](https://github.com/NormXU/ERNIE-Layout-Pytorch/blob/main/tools/convert2torch.py) script.
14
+ Please feel free to make any changes you need. For more details and use cases, please check the repo.
15
 
16
  **A Quick Example**
17
  ```python
 
63
  print(answer)
64
 
65
 
 
66
  ```